Kims12 commited on
Commit
8b664c8
·
verified ·
1 Parent(s): 51d3ed1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -147,7 +147,7 @@ def chatgpt_respond(
147
 
148
  try:
149
  response = openai.ChatCompletion.create(
150
- model="gpt-4o-mini", # 또는 다른 모델 ID 사용
151
  messages=messages,
152
  max_tokens=max_tokens,
153
  temperature=temperature,
@@ -178,24 +178,24 @@ def claude_respond(
178
  api_key=anthropic_api_key,
179
  )
180
 
181
- # 대화 기록을 하나의 프롬프트 문자열로 변환
182
- prompt = f"{system_message}\n"
183
  for human, assistant in chat_history:
184
  if human:
185
- prompt += f"Human: {human}\n"
186
  if assistant:
187
- prompt += f"Assistant: {assistant}\n"
188
- prompt += f"Human: {message}\nAssistant:"
189
 
190
  try:
191
- response = client.completions.create(
192
- model="claude-3-haiku-20240307",
193
- prompt=prompt,
194
- max_tokens_to_sample=max_tokens,
195
  temperature=temperature,
196
  top_p=top_p,
 
197
  )
198
- assistant_message = response.completion
199
  chat_history.append((message, assistant_message))
200
  return chat_history
201
  except Exception as e:
 
147
 
148
  try:
149
  response = openai.ChatCompletion.create(
150
+ model="gpt-4", # 올바른 모델 ID 수정 (예: "gpt-4")
151
  messages=messages,
152
  max_tokens=max_tokens,
153
  temperature=temperature,
 
178
  api_key=anthropic_api_key,
179
  )
180
 
181
+ # 대화 기록을 하나의 메시지 리스트로 변환
182
+ messages = [{"role": "system", "content": system_message}]
183
  for human, assistant in chat_history:
184
  if human:
185
+ messages.append({"role": "user", "content": human})
186
  if assistant:
187
+ messages.append({"role": "assistant", "content": assistant})
188
+ messages.append({"role": "user", "content": message})
189
 
190
  try:
191
+ response = client.messages.create(
192
+ model="claude-3.0", # 사용 가능한 Claude 모델로 수정
193
+ max_tokens=max_tokens,
 
194
  temperature=temperature,
195
  top_p=top_p,
196
+ messages=messages,
197
  )
198
+ assistant_message = response['completion']
199
  chat_history.append((message, assistant_message))
200
  return chat_history
201
  except Exception as e: