Add support for queryig code interpreter

superagent-ai · Feb 28, 2024 · 8f3aaee · 8f3aaee
1 parent 8264094
commit 8f3aaee
Show file tree

Hide file tree

Showing 5 changed files with 302 additions and 263 deletions.
diff --git a/README.md b/README.md
@@ -49,7 +49,8 @@ Input example:
     "encoder": {
         "type": "openai",
         "name": "text-embedding-3-small",
-    }
+    },
+    "interpreter_mode": False, # Set to True if you wish to run computation Q&A with a code interpreter
     "session_id": "my_session_id" # keeps micro-vm sessions and enables caching 
 }
 ```

diff --git a/models/query.py b/models/query.py
@@ -12,6 +12,7 @@ class RequestPayload(BaseModel):
     index_name: str
     encoder: Encoder
     session_id: Optional[str] = None
+    interpreter_mode: Optional[bool] = False
 
 
 class ResponseData(BaseModel):