File size: 1,107 Bytes
b23964e 85696fa 71467d9 43033c2 71467d9 43033c2 71467d9 43033c2 71467d9 6fadd30 71467d9 43033c2 3e00ca9 43033c2 71467d9 43033c2 71467d9 f322336 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import gradio as gr
import requests
import os
API_KEY = os.environ.get("TOGETHER_API_KEY")
MODEL_NAME = "mistralai/Mixtral-8x7B-Instruct-v0.1"
def ask_ai(message, history):
if not API_KEY:
return "API key not found. Please set it in Settings → Secrets."
url = "https://api.together.xyz/v1/chat/completions"
headers = {
"Authorization": f"Bearer {API_KEY}",
"Content-Type": "application/json"
}
messages = [{"role": "system", "content": "You are Vedanta AI, a helpful assistant."}]
for user, bot in history:
messages.append({"role": "user", "content": user})
messages.append({"role": "assistant", "content": bot})
messages.append({"role": "user", "content": message})
data = {
"model": MODEL_NAME,
"messages": messages
}
try:
response = requests.post(url, headers=headers, json=data)
response.raise_for_status()
result = response.json()
reply = result['choices'][0]['message']['content']
return reply
except Exception as e:
return f" Error: {str(e)}"
|