import os
import django
from django.conf import settings

# Setup Django environment
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings')
django.setup()

from ai_agent.gemini_client import GeminiClient
from ai_agent.rag_service import RAGService

def test_integration():
    print("Testing Gemini Client...")
    try:
        client = GeminiClient()
        response = client.generate_content("Hello, simply say 'AI is working'")
        print(f"Gemini Response: {response}")
    except Exception as e:
        print(f"Gemini Client Failed: {e}")

    print("\nTesting RAG Service (Vector DB)...")
    try:
        rag = RAGService()
        rag.add_document("This is a test document for the AI knowledge base.", {"source": "test"})
        print("Document added.")
        
        results = rag.search("test document")
        print(f"Search Results: {len(results)}")
        for r in results:
            print(f" - {r['content']} (Score: {r['score']})")
            
    except Exception as e:
        print(f"RAG Service Failed (Expected if docker container is not running): {e}")

if __name__ == "__main__":
    test_integration()