fix deps

Author: gkorland

api/llm.py

@@ -2,12 +2,6 @@
 import logging
 
 from graphrag_sdk.models.openai import OpenAiGenerativeModel
-#from graphrag_sdk.models.gemini import GeminiGenerativeModel
-from prompts import (CYPHER_GEN_SYSTEM,
-                     CYPHER_GEN_PROMPT,
-                     GRAPH_QA_SYSTEM,
-                     GRAPH_QA_PROMPT,
-                    )
 from graphrag_sdk import (
     Ontology,
     Entity,
@@ -18,6 +12,14 @@
     KnowledgeGraphModelConfig
 )
 
+#from graphrag_sdk.models.gemini import GeminiGenerativeModel
+from .prompts import (CYPHER_GEN_SYSTEM,
+                     CYPHER_GEN_PROMPT,
+                     GRAPH_QA_SYSTEM,
+                     GRAPH_QA_PROMPT,
+                    )
+
+
 # Configure logging
 logging.basicConfig(level=logging.DEBUG, format='%(filename)s - %(asctime)s - %(levelname)s - %(message)s')
 

poetry.lock

@@ -15,7 +15,6 @@ tree-sitter-c = "^0.21.4"
 tree-sitter-python = "^0.21.0"
 flask = "^3.0.3"
 python-dotenv = "^1.0.1"
-prompts = "^0.0.1"
 
 [tool.poetry.group.test.dependencies]
 pytest = "^8.2.0"

pyproject.toml

@@ -44,7 +44,6 @@ parso==0.8.4 ; python_version >= "3.9" and python_version < "4.0"
 pexpect==4.9.0 ; python_version >= "3.9" and python_version < "4.0" and sys_platform != "win32"
 platformdirs==4.3.6 ; python_version >= "3.9" and python_version < "4.0"
 prompt-toolkit==3.0.48 ; python_version >= "3.9" and python_version < "4.0"
-prompts==0.0.1 ; python_version >= "3.9" and python_version < "4.0"
 psutil==6.1.0 ; python_version >= "3.9" and python_version < "4.0"
 ptyprocess==0.7.0 ; python_version >= "3.9" and python_version < "4.0" and sys_platform != "win32"
 pure-eval==0.2.3 ; python_version >= "3.9" and python_version < "4.0"