[ Fix ] for the failing embedchain tests (#1899)

mem0ai · Sep 25, 2024 · 44ee48e · 44ee48e
1 parent 5525c4e
commit 44ee48e
Show file tree

Hide file tree

Showing 4 changed files with 71 additions and 29 deletions.
diff --git a/embedchain/embedchain/config/model_prices_and_context_window.json b/embedchain/embedchain/config/model_prices_and_context_window.json
@@ -13,14 +13,14 @@
         "input_cost_per_token": 0.000005,
         "output_cost_per_token": 0.000015
     },
-   "gpt-4o-mini": {
+   "openai/gpt-4o-mini": {
         "max_tokens": 4096,
         "max_input_tokens": 128000,
         "max_output_tokens": 4096,
         "input_cost_per_token": 0.00000015,
         "output_cost_per_token": 0.00000060
     },
-    "gpt-4o-mini-2024-07-18": {
+    "openai/gpt-4o-mini-2024-07-18": {
         "max_tokens": 4096,
         "max_input_tokens": 128000,
         "max_output_tokens": 4096,

diff --git a/embedchain/poetry.lock b/embedchain/poetry.lock
diff --git a/embedchain/pyproject.toml b/embedchain/pyproject.toml
@@ -100,7 +100,7 @@ chromadb = "^0.4.24"
 posthog = "^3.0.2"
 rich = "^13.7.0"
 beautifulsoup4 = "^4.12.2"
-pypdf = "^4.0.1"
+pypdf = "^5.0.0"
 gptcache = "^0.1.43"
 pysbd = "^0.3.4"
 mem0ai = "^0.1.15"
@@ -139,6 +139,7 @@ alembic = "^1.13.1"
 langchain-cohere = "^0.1.4"
 langchain-community = "^0.2.6"
 langchain-aws = {version = "^0.1.10", optional = true}
+langsmith = "^0.1.17"
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.3.0"

diff --git a/embedchain/tests/llm/test_openai.py b/embedchain/tests/llm/test_openai.py
@@ -85,7 +85,7 @@ def test_get_llm_model_answer_with_token_usage(config, mocker):
         "prompt_tokens": 1,
         "completion_tokens": 2,
         "total_tokens": 3,
-        "total_cost": 5.5e-06,
+        "total_cost": 1.35e-06,
         "cost_currency": "USD",
     }
     mocked_get_answer.assert_called_once_with("Test query", test_config)