Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -140,7 +140,13 @@ with gr.Blocks(css_paths="app.css") as demo:
|
|
| 140 |
_history = []
|
| 141 |
|
| 142 |
messages = history_to_messages(_history, _setting['system'])
|
| 143 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 144 |
|
| 145 |
try:
|
| 146 |
# μ€κ° μν νμ
|
|
@@ -153,14 +159,14 @@ with gr.Blocks(css_paths="app.css") as demo:
|
|
| 153 |
response = client.messages.create(
|
| 154 |
model="claude-3-sonnet-20240229",
|
| 155 |
max_tokens=4096,
|
| 156 |
-
|
| 157 |
-
|
| 158 |
-
for msg in messages
|
| 159 |
-
]
|
| 160 |
)
|
| 161 |
|
| 162 |
content = response.content[0].text
|
| 163 |
-
_history = messages_to_history(
|
|
|
|
|
|
|
| 164 |
'role': Role.ASSISTANT,
|
| 165 |
'content': content
|
| 166 |
}])
|
|
|
|
| 140 |
_history = []
|
| 141 |
|
| 142 |
messages = history_to_messages(_history, _setting['system'])
|
| 143 |
+
system_message = messages[0]['content'] # Get system message
|
| 144 |
+
|
| 145 |
+
# Convert remaining messages to Claude format (excluding system message)
|
| 146 |
+
claude_messages = [
|
| 147 |
+
{"role": msg["role"] if msg["role"] != "system" else "user", "content": msg["content"]}
|
| 148 |
+
for msg in messages[1:] + [{'role': Role.USER, 'content': query}]
|
| 149 |
+
]
|
| 150 |
|
| 151 |
try:
|
| 152 |
# μ€κ° μν νμ
|
|
|
|
| 159 |
response = client.messages.create(
|
| 160 |
model="claude-3-sonnet-20240229",
|
| 161 |
max_tokens=4096,
|
| 162 |
+
system=system_message, # System message as separate parameter
|
| 163 |
+
messages=claude_messages
|
|
|
|
|
|
|
| 164 |
)
|
| 165 |
|
| 166 |
content = response.content[0].text
|
| 167 |
+
_history = messages_to_history([
|
| 168 |
+
{'role': Role.SYSTEM, 'content': system_message}
|
| 169 |
+
] + claude_messages + [{
|
| 170 |
'role': Role.ASSISTANT,
|
| 171 |
'content': content
|
| 172 |
}])
|