fix import

gkorland · gkorland · commit 24abb4e4409b · 2024-12-30T16:48:41.000+02:00
diff --git a/api/llm.py b/api/llm.py
@@ -1,7 +1,7 @@
 import os
 import logging
 
-from graphrag_sdk.models.openai import OpenAiGenerativeModel
+from graphrag_sdk.models.litellm import LiteModel
 from graphrag_sdk import (
     Ontology,
     Entity,
