Merge branch 'feature/cog-1036-improving-visualization-endpoint' of h…

…ttps://github.com/topoteretes/cognee into feature/cog-1036-improving-visualization-endpoint
topoteretes · Feb 6, 2025 · c3f5dec · c3f5dec
2 parents f24506f + 4e3a90b
commit c3f5dec
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 108 deletions.
diff --git a/.github/workflows/reusable_python_example.yml b/.github/workflows/reusable_python_example.yml
@@ -16,24 +16,10 @@ on:
         required: true
       GRAPHISTRY_PASSWORD:
         required: true
-      LLM_MODEL:
-        required: true
-      LLM_ENDPOINT:
-        required: true
       LLM_API_KEY:
         required: true
       OPENAI_API_KEY:
         required: false
-      LLM_API_VERSION:
-        required: true
-      EMBEDDING_MODEL:
-        required: true
-      EMBEDDING_ENDPOINT:
-        required: true
-      EMBEDDING_API_KEY:
-        required: true
-      EMBEDDING_API_VERSION:
-        required: true
 
 env:
   RUNTIME__LOG_LEVEL: ERROR
@@ -70,15 +56,8 @@ jobs:
         env:
           ENV: 'dev'
           PYTHONFAULTHANDLER: 1
-          LLM_MODEL: ${{ secrets.LLM_MODEL }}
-          LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }}
-          LLM_API_KEY: ${{ secrets.LLM_API_KEY }}
-          LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }}
+          LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
-          EMBEDDING_MODEL: ${{ secrets.EMBEDDING_MODEL }}
-          EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }}
-          EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }}
-          EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }}
           GRAPHISTRY_USERNAME: ${{ secrets.GRAPHISTRY_USERNAME }}
           GRAPHISTRY_PASSWORD: ${{ secrets.GRAPHISTRY_PASSWORD }}
         run: poetry run python ${{ inputs.example-location }} ${{ inputs.arguments }}
diff --git a/.github/workflows/test_code_graph_example.yml b/.github/workflows/test_code_graph_example.yml
@@ -17,13 +17,7 @@ jobs:
         example-location: ./examples/python/code_graph_example.py
         arguments: "--repo_path ./evals"
       secrets:
-        LLM_MODEL: ${{ secrets.LLM_MODEL }}
-        LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }}
-        LLM_API_KEY: ${{ secrets.LLM_API_KEY }}
-        LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }}
-        EMBEDDING_MODEL: ${{ secrets.EMBEDDING_MODEL }}
-        EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }}
-        EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }}
-        EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }}
+        OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+        LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }}
         GRAPHISTRY_USERNAME: ${{ secrets.GRAPHISTRY_USERNAME }}
         GRAPHISTRY_PASSWORD: ${{ secrets.GRAPHISTRY_PASSWORD }}
diff --git a/.github/workflows/test_multimedia_example.yaml b/.github/workflows/test_multimedia_example.yaml
@@ -16,13 +16,7 @@ jobs:
       with:
         example-location: ./examples/python/multimedia_example.py
       secrets:
-        #LLM_MODEL: ${{ secrets.LLM_MODEL }}
-        #LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }}
         LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} # Use OpenAI until we deploy models to handle multimedia
-        #LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }}
-        EMBEDDING_MODEL: ${{ secrets.EMBEDDING_MODEL }}
-        EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }}
-        EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }}
-        EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }}
+        OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
         GRAPHISTRY_USERNAME: ${{ secrets.GRAPHISTRY_USERNAME }}
         GRAPHISTRY_PASSWORD: ${{ secrets.GRAPHISTRY_PASSWORD }}
diff --git a/cognee-mcp/pyproject.toml b/cognee-mcp/pyproject.toml
@@ -21,8 +21,5 @@ build-backend = "hatchling.build"
 [tool.hatch.build.targets.wheel]
 packages = ["src"]
 
-[tool.uv.sources]
-cognee = { path = "../../cognee" }
-
 [project.scripts]
 cognee = "src:main"
diff --git a/cognee-mcp/uv.lock b/cognee-mcp/uv.lock
diff --git a/cognee/tasks/repo_processor/get_local_dependencies.py b/cognee/tasks/repo_processor/get_local_dependencies.py
@@ -73,6 +73,12 @@ def _update_code_entity(script: jedi.Script, code_entity: Dict[str, any]) -> Non
         # TODO: See if there is a way to handle EOFError properly
         logger.error(f"Failed to analyze code entity {code_entity['name']}: {e}")
         return
+    except AttributeError as e:
+        # TODO: See if there is a way to handle AttributeError properly
+        logger.error(f"Failed to analyze code entity {code_entity['name']}: {e}")
+    except AssertionError as e:
+        # TODO: See if there is a way to handle AttributeError properly
+        logger.error(f"Failed to analyze code entity {code_entity['name']}: {e}")
     except Exception as e:
         # logging.warning(f"Failed to analyze code entity {code_entity['name']}: {e}")
         logger.error(f"Failed to analyze code entity {code_entity['name']}: {e}")