Deploy CodeMode via Agent

app.py

@@ -143,23 +143,28 @@ def get_chunks_for_file_baseline(file_name):
         return {"error": "No file selected"}
     
     try:
+        # Get all data and filter in Python
         data = baseline_collection.get(
-            where={"file_name": file_name},
             include=["documents", "metadatas", "embeddings"]
         )
         
         if not data['documents']:
-            return {"error": "No chunks found for this file"}
+            return {"error": "No chunks found"}
         
+        # Filter by file_name in Python
         chunks = []
-        for i, (doc, meta) in enumerate(zip(data['documents'], data['metadatas'])):
-            chunks.append({
-                "chunk_id": i + 1,
-                "content": doc[:500] + "..." if len(doc) > 500 else doc,
-                "full_length": len(doc),
-                "metadata": meta,
-                "embedding_dim": len(data['embeddings'][i]) if data['embeddings'] else 0
-            })
+        for i, (doc, meta, emb) in enumerate(zip(data['documents'], data['metadatas'], data['embeddings'])):
+            if meta.get("file_name") == file_name:
+                chunks.append({
+                    "chunk_id": len(chunks) + 1,
+                    "content": doc[:500] + "..." if len(doc) > 500 else doc,
+                    "full_length": len(doc),
+                    "metadata": meta,
+                    "embedding_dim": len(emb) if emb else 0
+                })
+        
+        if not chunks:
+            return {"error": "No chunks found for this file"}
         
         return {
             "file_name": file_name,
@@ -175,23 +180,28 @@ def get_chunks_for_file_finetuned(file_name):
         return {"error": "No file selected"}
     
     try:
+        # Get all data and filter in Python
         data = finetuned_collection.get(
-            where={"file_name": file_name},
             include=["documents", "metadatas", "embeddings"]
         )
         
         if not data['documents']:
-            return {"error": "No chunks found for this file"}
+            return {"error": "No chunks found"}
         
+        # Filter by file_name in Python
         chunks = []
-        for i, (doc, meta) in enumerate(zip(data['documents'], data['metadatas'])):
-            chunks.append({
-                "chunk_id": i + 1,
-                "content": doc[:500] + "..." if len(doc) > 500 else doc,
-                "full_length": len(doc),
-                "metadata": meta,
-                "embedding_dim": len(data['embeddings'][i]) if data['embeddings'] else 0
-            })
+        for i, (doc, meta, emb) in enumerate(zip(data['documents'], data['metadatas'], data['embeddings'])):
+            if meta.get("file_name") == file_name:
+                chunks.append({
+                    "chunk_id": len(chunks) + 1,
+                    "content": doc[:500] + "..." if len(doc) > 500 else doc,
+                    "full_length": len(doc),
+                    "metadata": meta,
+                    "embedding_dim": len(emb) if emb else 0
+                })
+        
+        if not chunks:
+            return {"error": "No chunks found for this file"}
         
         return {
             "file_name": file_name,