code-slicer commited on
Commit
e0c4f9d
·
verified ·
1 Parent(s): c5676e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -261,6 +261,13 @@ def call_ollama_stream(messages, *, model: str = OLLAMA_MODEL,
261
  if read_timeout_sec is None:
262
  read_timeout_sec = int(os.getenv("OLLAMA_TIMEOUT", "300")) # ✅ 5분
263
 
 
 
 
 
 
 
 
264
  payload = {
265
  "model": model,
266
  "messages": _msgs, # <- 기존 그대로
 
261
  if read_timeout_sec is None:
262
  read_timeout_sec = int(os.getenv("OLLAMA_TIMEOUT", "300")) # ✅ 5분
263
 
264
+ url = f"{OLLAMA_HOST}/api/chat"
265
+
266
+ _msgs = []
267
+ if system_prompt:
268
+ _msgs.append({"role": "system", "content": system_prompt})
269
+ _msgs.extend(messages)
270
+
271
  payload = {
272
  "model": model,
273
  "messages": _msgs, # <- 기존 그대로