feat: update deps, fix stats

shopware · Jan 31, 2024 · 1f4ffbe · 1f4ffbe
1 parent febe770
commit 1f4ffbe
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 35 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,9 +12,9 @@ langchain = "^0.0.354"
 faiss-cpu = "^1.7.4"
 unstructured = "^0.11.8"
 libmagic = "^1.0"
-openai = "^1.6.1"
+openai = "^0.28.1"
 tiktoken = "^0.5.2"
-fastapi = "^0.108.0"
+fastapi = "^0.109.0"
 pydantic = "^2.5.3"
 pydantic-settings = "^2.1.0"
 uvicorn = "^0.25.0"

diff --git a/web/answering.py b/web/answering.py
@@ -83,8 +83,13 @@ async def generate_answer(question: str, collection):
         # track event
         await send_event('all', 'qa', {**{"question": question, "collection": collection, "instances": instances}, **cb.__dict__})
 
-        output['stats'] = cb
-
+        output['stats'] = {
+            'total_cost': cb.total_cost,
+            'total_tokens': cb.total_tokens,
+            'prompt_tokens': cb.prompt_tokens,
+            'completion_tokens': cb.completion_tokens,
+            'successful_requests': cb.successful_requests,
+        }
     return output