jeongsoo commited on
Commit
5243fd2
ยท
1 Parent(s): b12512e

Add application file

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -50,11 +50,20 @@ except Exception as e:
50
  self.vector_store = vector_store
51
 
52
  # ํ™˜๊ฒฝ ์„ค์ • ์ž„ํฌํŠธ
53
- from config import OPENAI_API_KEY, LLM_MODEL, USE_OPENAI, TOP_K_RETRIEVAL, OLLAMA_HOST
 
 
 
 
 
 
 
 
 
54
 
55
  try:
56
  # LLM ์ดˆ๊ธฐํ™”
57
- if USE_OPENAI:
58
  self.llm = ChatOpenAI(
59
  model_name=LLM_MODEL,
60
  temperature=0.2,
 
50
  self.vector_store = vector_store
51
 
52
  # ํ™˜๊ฒฝ ์„ค์ • ์ž„ํฌํŠธ
53
+ from config import OPENAI_API_KEY, LLM_MODEL, USE_OPENAI, TOP_K_RETRIEVAL, IS_HUGGINGFACE
54
+
55
+ # OLLAMA_HOST๋Š” HuggingFace ํ™˜๊ฒฝ์—์„œ๋Š” ์ •์˜๋˜์ง€ ์•Š์œผ๋ฏ€๋กœ ์กฐ๊ฑด๋ถ€ ๊ฐ€์ ธ์˜ค๊ธฐ
56
+ OLLAMA_HOST = "http://localhost:11434" # ๊ธฐ๋ณธ๊ฐ’
57
+ if not IS_HUGGINGFACE and not USE_OPENAI:
58
+ try:
59
+ from config import OLLAMA_HOST
60
+ print(f"Ollama ํ˜ธ์ŠคํŠธ: {OLLAMA_HOST}")
61
+ except ImportError:
62
+ print("OLLAMA_HOST ์„ค์ •์„ ์ฐพ์„ ์ˆ˜ ์—†์–ด ๊ธฐ๋ณธ๊ฐ’ ์‚ฌ์šฉ")
63
 
64
  try:
65
  # LLM ์ดˆ๊ธฐํ™”
66
+ if USE_OPENAI or IS_HUGGINGFACE:
67
  self.llm = ChatOpenAI(
68
  model_name=LLM_MODEL,
69
  temperature=0.2,