Merge branch 'main' into gkorland-patch-2

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')