Merge pull request #31 from ProspektStudio/david

Disable CAG

Author: dtom90

.gitignore

@@ -1,14 +1,14 @@
 # dependencies
-/node_modules
+client/node_modules
 /.pnp
 .pnp.js
 
 # testing
 /coverage
 
 # next.js
-/.next/
-/out/
+client/.next/
+client/out/
 
 # python
 __pycache__

api/main.py

@@ -46,15 +46,15 @@ async def get_satellite_info_rag(
     )
 
 
-@app.get("/api/satellite-info-cag")
-async def get_satellite_info_cag(
-    group: str = Query(..., min_length=1, max_length=50),
-    name: str = Query(..., min_length=1, max_length=50)
-):
-    return StreamingResponse(
-        generate_satellite_info_stream(name, cag_content_stream),
-        media_type="text/event-stream"
-    )
+# @app.get("/api/satellite-info-cag")
+# async def get_satellite_info_cag(
+#     group: str = Query(..., min_length=1, max_length=50),
+#     name: str = Query(..., min_length=1, max_length=50)
+# ):
+#     return StreamingResponse(
+#         generate_satellite_info_stream(name, cag_content_stream),
+#         media_type="text/event-stream"
+#     )
 
 
 async def generate_satellite_info_stream(name: str, content_stream_func: Callable[[str], AsyncGenerator[str, None]]):

api/models/__init__.py

@@ -55,13 +55,15 @@ async def rag_content_stream(name: str):
         if chunk:
             yield chunk
 
-cag_satellite_agent = CagAgent(
-    base_prompt=BASE_PROMPT,
-    llm_model=GEMINI_MODEL,
-)
-
-async def cag_content_stream(name: str):
-    prompt = generate_prompt(name)
-    async for chunk in cag_satellite_agent.ask(prompt):
-        if chunk:
-            yield chunk
+# CAG Agent Config
+
+# cag_satellite_agent = CagAgent(
+#     base_prompt=BASE_PROMPT,
+#     llm_model=GEMINI_MODEL,
+# )
+
+# async def cag_content_stream(name: str):
+#     prompt = generate_prompt(name)
+#     async for chunk in cag_satellite_agent.ask(prompt):
+#         if chunk:
+#             yield chunk

api/models/cag_agent.py

@@ -7,7 +7,7 @@
 from google.genai import types
 from .doc_loader import load_webpages
 
-logger = logging.getLogger(__name__)
+logger = logging.getLogger("uvicorn")
 
 LLM_TEMPERATURE = 0.0
 

.env.development.local client/.env.development.local.env.development.local renamed to client/.env.development.local

@@ -26,10 +26,10 @@ const agentInfo: Record<string, { name: string; description: string }> = {
     name: 'Retrieval Augmented Generation',
     description: "Retrieval Augmented Generation (RAG) implementation uses Google's Vertex AI for embeddings and Gemini 2.0 Flash as the LLM, with LangChain providing the framework. The system loads satellite-related documents, processes them into chunks, stores them in an in-memory vector store, and retrieves relevant information and generates satellite information.",
   },
-  [Agent.CAG]: {
-    name: 'Cache-Augmented Generation',
-    description: "Cache-Augmented Generation (CAG) implementation uses caching to store satellite-related documents and retrieve relevant information and generates satellite information.",
-  },
+  // [Agent.CAG]: {
+  //   name: 'Cache-Augmented Generation',
+  //   description: "Cache-Augmented Generation (CAG) implementation uses caching to store satellite-related documents and retrieve relevant information and generates satellite information.",
+  // },
 }
 
 const AiInfo: React.FC<AiInfoProps> = ({ selectedSatellite }) => {