|
|
| import os |
| import gradio as gr |
| from smolagents import CodeAgent, InferenceClientModel |
| from smolagents import DuckDuckGoSearchTool |
|
|
| |
| def create_agent(): |
|
|
| token = os.getenv("HF_TOKEN") |
|
|
| model = InferenceClientModel( |
| model_id="Qwen/Qwen2.5-Coder-32B-Instruct", |
| token=token |
| ) |
|
|
| search_tool = DuckDuckGoSearchTool() |
|
|
| agent = CodeAgent( |
| tools=[search_tool], |
| model=model, |
| max_steps=6 |
| ) |
|
|
| return agent |
|
|
|
|
| |
| agent = create_agent() |
|
|
|
|
| |
| def chat(message, history): |
|
|
| try: |
| response = agent.run(message, reset=True) |
| return str(response) |
|
|
| except Exception as e: |
| return f"Error: {str(e)}" |
|
|
|
|
| |
| demo = gr.ChatInterface( |
| fn=chat, |
| title="Alfred - AI Research Agent", |
| description="Ask Alfred questions. He can search the web and answer using AI." |
| ) |
|
|
| if __name__ == "__main__": |
| demo.launch() |
|
|
