dev-yuje commited on
Commit
eb29e6d
Β·
1 Parent(s): 2188b59

fix: lazy initialize OpenAI models to prevent import-time collection crash in CI/CD

Browse files
Files changed (1) hide show
  1. src/retrieval/finRetrieval.py +4 -3
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(