Update agent_langchain.py

agent_langchain.py

@@ -13,7 +13,7 @@ from langchain.memory import ConversationBufferMemory
 # Environment & URLs
 # -------------------------------
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
-GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent"
+GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent"
 ROUTING_URL = os.environ.get("ROUTING_URL")  # Space 2 URL
 SPACE_URL = os.environ.get("SPACE_URL", "http://localhost:7860")
 
@@ -53,9 +53,9 @@ clf_model = AutoModelForSequenceClassification.from_pretrained(clf_model_name)
 chroma_client = chromadb.PersistentClient(path="/tmp/chroma")
 
 # ✅ Create or get your KB collection
-kb_collection = chroma_client.get_or_create_collection("KB")
+kb_collection = chroma_client.get_or_create_collection("Knowledge_Base")
 
-COLLECTION_NAME = "KB"
+COLLECTION_NAME = "Knowledge_Base"
 try:
     kb_collection = chroma_client.get_collection(COLLECTION_NAME)
 except: