v0.1.8.4

.gitignore

@@ -14,4 +14,4 @@ src/*.old.py
 src/toolSelector.py
 test/__pycache__/*.pyc
 src/baseInfra/memoryManager.py
-models/bge-large-en-v1.5/*.bin
+models/BAAI_bge-large-en-v1.5/*.bin

src/chroma_intf.py

@@ -144,8 +144,9 @@ def addText(inStr:str,metadata):
     md=meta_defaults
     for key in metadata.keys():
         md[key]=metadata[key]
-    md['timestamp']=datetime.now().strftime("%Y-%m-%d %H:%M:%S::%f")
-    md['ID']=datetime.now().strftime("%Y-%m-%d %H:%M:%S::%f")+"-conversation"
+    if "timestamp" not in metadata.keys():
+        md['timestamp']=datetime.now()
+    md['ID']=md['timestamp'].strftime("%Y-%m-%d %H:%M:%S::%f")+"-conversation"
 
     docs = [
         Document(page_content=inStr, metadata=md)]

src/main.py

@@ -10,7 +10,7 @@ from fastapi import FastAPI as Response
 from sse_starlette.sse import EventSourceResponse
 from starlette.responses import StreamingResponse
 from starlette.requests import Request
-from pydantic import BaseModel
+from pydantic import BaseModel, Extra
 from enum import Enum
 from typing import List, Dict, Any, Generator, Optional, cast, Callable
 from chroma_intf import *
@@ -19,9 +19,15 @@ import baseInfra.dropbox_handler as dbh
 class PathRequest(BaseModel):
     dir: str = "/"
 
+class MetaD(BaseModel):
+    timestamp: Optional[datetime]= datetime.now()
+    class Config:
+        allow_population_by_field_name = True
+        extra = Extra.allow
+
 class DocWithMeta(BaseModel):
     text: str = ""
-    metadata: Optional[dict] = {}
+    metadata: Optional[MetaD] = MetaD()
 
 async def catch_exceptions_middleware(
     request: Request, call_next: Callable[[Request], Any]