Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 # ------------------------------
 # Imports & Dependencies
 # ------------------------------
@@ -12,12 +10,28 @@ from langgraph.prebuilt import ToolNode
 from langgraph.graph.message import add_messages
 from typing_extensions import TypedDict, Annotated
 from typing import Sequence
+import chromadb
+from chromadb.config import Settings
 import re
 import os
 import streamlit as st
 import requests
 from langchain.tools.retriever import create_retriever_tool
 
+# Create directory for Chroma persistence
+os.makedirs("chroma_db", exist_ok=True)
+
+# ------------------------------
+# ChromaDB Client Configuration
+# ------------------------------
+chroma_client = chromadb.Client(
+    Settings(
+        chroma_db_impl="duckdb+parquet",
+        persist_directory="chroma_db",
+        anonymized_telemetry=False
+    )
+)
+
 # ------------------------------
 # Dummy Data: Research & Development Texts
 # ------------------------------
@@ -47,16 +61,21 @@ embeddings = OpenAIEmbeddings(
     model="text-embedding-3-large",
     # dimensions=1024  # Uncomment if needed
 )
+
 research_vectorstore = Chroma.from_documents(
     documents=research_docs,
     embedding=embeddings,
+    client=chroma_client,
     collection_name="research_collection"
 )
+
 development_vectorstore = Chroma.from_documents(
     documents=development_docs,
     embedding=embeddings,
+    client=chroma_client,
     collection_name="development_collection"
 )
+
 research_retriever = research_vectorstore.as_retriever()
 development_retriever = development_vectorstore.as_retriever()
 
@@ -256,7 +275,7 @@ def custom_tools_condition(state: AgentState):
     return END
 
 # ------------------------------
-# Workflow Configuration using LangGraph (Corrected)
+# Workflow Configuration using LangGraph
 # ------------------------------
 workflow = StateGraph(AgentState)