v0.1.8.0

src/chroma_intf.py

@@ -126,7 +126,7 @@ def getRelevantDocs(query:str):
             v="Info:"+item['page_content']+" "
             for key in item.metadata.keys():
                 if key != "ID":
-                    v+=key+":"+item.metadata[key]+" "
+                    v+=key+":"+str(item.metadata[key])+" "
             value.append(v)
         db_interface.add_to_cache(input=query,value=value)
     except:
@@ -134,7 +134,7 @@ def getRelevantDocs(query:str):
             v="Info:"+item.page_content+" "
             for key in item.metadata.keys():
                 if key != "ID":
-                    v+=key+":"+item.metadata[key]+" "
+                    v+=key+":"+str(item.metadata[key])+" "
             value.append(v)
         db_interface.add_to_cache(input=query,value=value)
     return retVal

src/main.py

@@ -16,7 +16,12 @@ from typing import List, Dict, Any, Generator, Optional, cast, Callable
 from chroma_intf import *
 import baseInfra.dropbox_handler as dbh
 
+class PathRequest(BaseModel):
+    dir: str = "/"
 
+class DocWithMeta(BaseModel):
+    text: str = ""
+    metadata: Optional[dict] = {}
 
 async def catch_exceptions_middleware(
     request: Request, call_next: Callable[[Request], Any]
@@ -49,11 +54,12 @@ async def get_matching_docs(inStr: str ) -> Any:
     return getRelevantDocs(inStr)
 
 @app.post(api_base+"/addTextDocument")
-async def add_text_document(inStr: str,metadata: dict ) -> Any:
+async def add_text_document(inDoc: DocWithMeta ) -> Any:
     """
      Add text and metadata to the db
     """
-    return addText(inStr,metadata)
+    
+    return addText(inDoc.text,inDoc.metadata)
 
 @app.get(api_base+"/persist")
 async def persist_db():
@@ -64,6 +70,19 @@ async def persist_db():
 async def reset_db():
     return await dbh.restoreFolder("db")
 
+@app.post(api_base+"/walk")
+def walk(path: PathRequest):
+    print("Received walk request")
+    dirs=[]
+    try:
+        for root, items, files in os.walk(path.dir,topdown=True):
+            for item in items:
+                dirs.append(item)
+    except Exception:
+        print("got exception",Exception)
+    response= JSONResponse(content= {"dirs":dirs})
+    return response
+
 print(__name__)
 
 if __name__ == '__main__' or __name__ == "src.main":