fix: lazy initialize OpenAI models to prevent import-time collection crash in CI/CD
Browse files
src/retrieval/finRetrieval.py
CHANGED
|
@@ -47,9 +47,6 @@ def get_neo4j_driver() -> neo4j.Driver:
|
|
| 47 |
return d
|
| 48 |
|
| 49 |
|
| 50 |
-
rag_llm = OpenAILLM(model_name="gpt-4o", model_params={"temperature": 0})
|
| 51 |
-
embedder = OpenAIEmbeddings(model="text-embedding-3-small")
|
| 52 |
-
|
| 53 |
INDEX_NAME = "content_vector_index"
|
| 54 |
|
| 55 |
# ββββββββββββββββββββββββββββββββββββββββββ
|
|
@@ -186,6 +183,10 @@ class LazyGraphRAG:
|
|
| 186 |
if self._graphrag is not None:
|
| 187 |
return
|
| 188 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 189 |
driver = get_neo4j_driver()
|
| 190 |
|
| 191 |
vector_retriever = VectorRetriever(
|
|
|
|
| 47 |
return d
|
| 48 |
|
| 49 |
|
|
|
|
|
|
|
|
|
|
| 50 |
INDEX_NAME = "content_vector_index"
|
| 51 |
|
| 52 |
# ββββββββββββββββββββββββββββββββββββββββββ
|
|
|
|
| 183 |
if self._graphrag is not None:
|
| 184 |
return
|
| 185 |
|
| 186 |
+
# OpenAI ν΄λΌμ΄μΈνΈ λ° μλ² λ μ§μ° μ΄κΈ°ν (CI ν¬λμ λ°©μ§)
|
| 187 |
+
rag_llm = OpenAILLM(model_name="gpt-4o", model_params={"temperature": 0})
|
| 188 |
+
embedder = OpenAIEmbeddings(model="text-embedding-3-small")
|
| 189 |
+
|
| 190 |
driver = get_neo4j_driver()
|
| 191 |
|
| 192 |
vector_retriever = VectorRetriever(
|