chaeya commited on
Commit
0f81b84
·
unverified ·
1 Parent(s): 67552c8

respond 함수에서 API 호출 로직을 주석 처리하고, 응답 처리 방식을 단순화함

Browse files
Files changed (1) hide show
  1. app.py +6 -11
app.py CHANGED
@@ -8,23 +8,18 @@ def respond(message, history):
8
  if history is None:
9
  history = []
10
 
 
11
  messages = history.copy()
 
12
  messages.append({"role": "user", "content": message})
13
 
14
- payload = {
15
- "model": MODEL_NAME,
16
- "messages": messages,
17
- "stream": False
18
- }
19
-
20
- response = requests.post(OLLAMA_API_URL, json=payload)
21
- response.raise_for_status()
22
-
23
- data = response.json()
24
- assistant_reply = data['choices'][0]['message']['content']
25
 
 
26
  messages.append({"role": "assistant", "content": assistant_reply})
27
 
 
28
  return assistant_reply, messages
29
 
30
  demo = gr.ChatInterface(respond, type="messages")
 
8
  if history is None:
9
  history = []
10
 
11
+ # history 복사
12
  messages = history.copy()
13
+ # 사용자 메시지 추가
14
  messages.append({"role": "user", "content": message})
15
 
16
+ # (예: API 호출)
17
+ assistant_reply = "API에서 받은 답변 텍스트"
 
 
 
 
 
 
 
 
 
18
 
19
+ # 어시스턴트 메시지 추가
20
  messages.append({"role": "assistant", "content": assistant_reply})
21
 
22
+ # 반드시 (응답 텍스트, 메시지 리스트) 반환
23
  return assistant_reply, messages
24
 
25
  demo = gr.ChatInterface(respond, type="messages")