Spaces:
Sleeping
Sleeping
File size: 1,527 Bytes
959296e c92c3bf d93dbdd 9b5b26a 0f61d7b 2764a1c 1c61c68 4c2a00d d4578f1 4db54a2 e55e264 67c8f73 7686479 ed36d14 95de6d7 9b5b26a 8f16109 95c49b7 826fb71 65b9821 6dc9c22 0f61d7b 95c49b7 0f61d7b 1cde8a3 aa945ae 95c49b7 aa945ae 0ae92d5 1cde8a3 0ae92d5 8c01ffb c3644fc 1c61c68 8c01ffb ed36d14 e55e264 85f8fc6 8c01ffb 8fe992b e55e264 8c01ffb 1795753 8c01ffb 3d41af7 d4578f1 6e73783 8fe992b 9b5b26a c114199 |
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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import sys # Add this import to fix the error
import datetime
import requests
from checks.failed_check import create_failed_gradio_ui
from checks.endpoint_check import check_public_endpoint
import os
import pytz # Had to give it permission in Code agent
from tools.tools import load_tools
from agents.model import load_huggingface_model
from agents.prompts import load_prompts
from Gradio_UI import GradioUI
# Prechecks
# Get the URI for the endpoint
endpoint_uri = load_huggingface_model()
# Test the endpoint
response = check_public_endpoint(endpoint_uri)
if response.status is False:
create_failed_gradio_ui()
else:
pass
'''
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id= endpoint_uri,
custom_role_conversions=None,
)
'''
'''
model = LiteLLMModel(
model_id="gemini/gemini-2.0-flash-exp",
max_tokens=2096,
temperature=0.6,
api_key=os.getenv("LITELLM_API_KEY")
)
'''
'''
# Load prompts
combined_prompts = load_prompts()
combined_tools = load_tools() ## from the tools folder (don't remove final answer)
agent = CodeAgent(
model=model,
tools=combined_tools,
max_steps=6,
verbosity_level=3,
grammar=None,
planning_interval=None,
name="Content Agent",
description="Evaluates whether text is polite or impolite. ",
prompt_templates=combined_prompts,
additional_authorized_imports=["pytz"]
)
GradioUI(agent).launch()
''' |