Merge branch 'main' into fix/rename-remaining-query-to-query-text-kwargs

topoteretes · Nov 18, 2024 · 64bc425 · 64bc425
2 parents 7a2fc61 + 34e140a
commit 64bc425
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/.github/workflows/test_cognee_llama_index_notebook.yml b/.github/workflows/test_cognee_llama_index_notebook.yml
@@ -46,7 +46,7 @@ jobs:
 
       - name: Install dependencies
         run: |
-          poetry install --no-interaction --all-extras --no-root
+          poetry install --no-interaction --all-extras 
           poetry add jupyter --no-interaction
 
       - name: Execute Jupyter Notebook

diff --git a/cognee-frontend/src/ui/Partials/SettingsModal/Settings.tsx b/cognee-frontend/src/ui/Partials/SettingsModal/Settings.tsx
@@ -30,8 +30,8 @@ const defaultProvider = {
 };
 
 const defaultModel = {
-  label: 'gpt-4o',
-  value: 'gpt-4o',
+  label: 'gpt-4o-mini',
+  value: 'gpt-4o-mini',
 };
 
 export default function Settings({ onDone = () => {}, submitButtonText = 'Save' }) {

diff --git a/cognee/infrastructure/llm/config.py b/cognee/infrastructure/llm/config.py
@@ -4,7 +4,7 @@
 
 class LLMConfig(BaseSettings):
     llm_provider: str = "openai"
-    llm_model: str = "gpt-4o"
+    llm_model: str = "gpt-4o-mini"
     llm_endpoint: str = ""
     llm_api_key: Optional[str] = None
     llm_temperature: float = 0.0

diff --git a/cognee/modules/settings/get_settings.py b/cognee/modules/settings/get_settings.py
@@ -73,6 +73,9 @@ def get_settings() -> SettingsDict:
             "providers": llm_providers,
             "models": {
                 "openai": [{
+                    "value": "gpt-4o-mini",
+                    "label": "gpt-4o-mini",
+                }, {
                     "value": "gpt-4o",
                     "label": "gpt-4o",
                 }, {