keynes42 commited on
Commit
662c1ac
·
verified ·
1 Parent(s): ce5cdd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- role = m['role'].value # "system" / "user" / "assistant"
 
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)