Update README and refactor backend and frontend code for improved cla…

…rity and functionality

README.md

@@ -15,7 +15,7 @@ DocQuery is a comprehensive project consisting of both frontend and backend comp
 1. Clone the repository:
 
     ```sh
-    git clone <repository-url>
+    git clone https://github.com/md-abid-hussain/docquery.git
     cd docquery-frontend
     ```
 

docquery-backend/app/agents/utils/database.py

@@ -4,7 +4,6 @@
 from .models import get_embedding_model
 
 client = MongoClient(os.environ.get("MONGODB_ATLAS_CLUSTER_URI"))
-
 DB_NAME = "docquery_db"
 COLLECTION_NAME = "vectorstore"
 ATLAS_VECTOR_SEARCH_INDEX = "rag_index"
@@ -18,10 +17,11 @@
     relevance_score_fn="cosine",
 )
 
+
 def get_vector_store():
     return vector_store
 
+
 def save_to_database(name: str):
     repository_collection = client["docquery"]["repositories"]
     repository_collection.insert_one({"name": name})
-

docquery-backend/app/main.py

@@ -11,17 +11,21 @@
 sdk = CopilotKitSDK(
     agents=[
         LangGraphAgent(
-            name="ingestion_agent", graph=ingestion_agent_graph, description="Ingestion Agent"
+            name="ingestion_agent",
+            graph=ingestion_agent_graph,
+            description="Ingestion Agent",
         ),
         LangGraphAgent(
-            name="qa_agent", graph=qa_agent_graph, description="QA Agent"
-        )
+            name="qa_agent",
+            graph=qa_agent_graph,
+            description="QA Agent"
+        ),
     ]
 )
 
 add_fastapi_endpoint(app, sdk, "/copilotkit")
 
 
-@app.get("/")
+@app.get("/test")
 def read_root():
-    return {"Hello": "World"}
+    return "running"

docquery-frontend/src/app/d/chat/owner/[owner]/repo/[repo]/page.tsx

@@ -2,14 +2,17 @@
 
 import { useParams } from 'next/navigation';
 import { useEffect } from 'react';
-import { useCoAgent, useCopilotReadable } from '@copilotkit/react-core';
+import { useCoAgent, useCopilotReadable, useCopilotMessagesContext, useCopilotContext } from '@copilotkit/react-core';
 import { CopilotChat } from '@copilotkit/react-ui';
 import { QAAgentState } from '@/lib/types';
 import useRepoDetails from '@/hooks/useRepoDetails';
 import CopilotTextMessage from '@/components/copilot/copilot-text-message';
 
 const ChatRepoPage = () => {
+
     const { owner, repo } = useParams();
+    const { setMessages } = useCopilotMessagesContext();
+    const { setAgentSession } = useCopilotContext()
 
     if (!owner || !repo) {
         return <div>Loading...</div>;
@@ -30,13 +33,18 @@ const ChatRepoPage = () => {
             repository_name: `${owner}/${repo}`,
             messages: []
         }
+
     });
 
     useEffect(() => {
         setState({
             question: "",
             repository_name: `${owner}/${repo}`,
         })
+        return () => {
+            setMessages([]);
+            setAgentSession(null);
+        }
     }, []);
 
     return (