| from mistralai import Mistral | |
| import gradio as gr | |
| import os | |
| from dotenv import load_dotenv | |
| import time | |
| load_dotenv() | |
| api_key = os.getenv("MISTRAL_API_KEY") | |
| client = Mistral(api_key=api_key) | |
| def echo(message, history): | |
| time.sleep(1.1) | |
| chat_response = client.agents.complete( | |
| agent_id="ag:4b4eaa5c:20250427:untitled-agent:27ff0afd", | |
| messages=[ | |
| { | |
| "role": "user", | |
| "content": message, | |
| } | |
| ], | |
| ) | |
| return chat_response.choices[0].message.content | |
| demo = gr.ChatInterface( | |
| fn=echo, | |
| type="messages", | |
| examples=["Hello, how's your day?", "I love the weather today"], | |
| title="Fined Tuned LLM with Reddit Comments", | |
| ) | |
| demo.launch() | |