Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import os, sys
|
|
|
2 |
import gradio as gr
|
3 |
import requests
|
4 |
import inspect
|
@@ -7,7 +8,6 @@ import pandas as pd
|
|
7 |
import torch, spaces
|
8 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
9 |
from smolagents import CodeAgent, HfApiModel
|
10 |
-
from smolagents.message import MessageRole
|
11 |
from huggingface_hub import InferenceClient, hf_hub_download
|
12 |
|
13 |
subprocess.run(["playwright", "install"], check=True)
|
@@ -111,7 +111,8 @@ class BasicAgent:
|
|
111 |
def _serialize_messages(self, messages):
|
112 |
prompt = []
|
113 |
for m in messages:
|
114 |
-
|
|
|
115 |
text = "".join([c['text'] for c in m['content']])
|
116 |
prompt.append(f"{role}: {text}")
|
117 |
return "\n".join(prompt)
|
|
|
1 |
import os, sys
|
2 |
+
from enum import Enum
|
3 |
import gradio as gr
|
4 |
import requests
|
5 |
import inspect
|
|
|
8 |
import torch, spaces
|
9 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
10 |
from smolagents import CodeAgent, HfApiModel
|
|
|
11 |
from huggingface_hub import InferenceClient, hf_hub_download
|
12 |
|
13 |
subprocess.run(["playwright", "install"], check=True)
|
|
|
111 |
def _serialize_messages(self, messages):
|
112 |
prompt = []
|
113 |
for m in messages:
|
114 |
+
r = m["role"]
|
115 |
+
role = r.value if isinstance(r, Enum) and hasattr(r, "value") else r # "system" / "user" / "assistant"
|
116 |
text = "".join([c['text'] for c in m['content']])
|
117 |
prompt.append(f"{role}: {text}")
|
118 |
return "\n".join(prompt)
|