Add application file
Browse files
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,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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,
|