Spaces:
Sleeping
Sleeping
File size: 1,521 Bytes
959296e c92c3bf d93dbdd 9b5b26a 0f61d7b 49198d7 1c61c68 4c2a00d d4578f1 4db54a2 56df9e5 67c8f73 03cdef6 ed36d14 95de6d7 03cdef6 9b5b26a 95c49b7 65b9821 e59746c 0f61d7b 95c49b7 0f61d7b 1cde8a3 aa945ae 95c49b7 aa945ae 0ae92d5 1cde8a3 0ae92d5 8c01ffb c3644fc 1c61c68 8c01ffb ed36d14 85f8fc6 8c01ffb 8fe992b 453152e 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 |
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 is_huggingface_endpoint, is_public_endpoint
import os
import pytz # Had to give it permission in Code agent
from tools.tools import tools
from agents.model import load_huggingface_mode
from agents.prompts import load_prompts
from Gradio_UI import GradioUI
# Get the URI for the endpoint
endpoint_uri = load_huggingface_model
# Test the endpoint
if not is_public_endpoint(endpoint_uri):
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()
agent = CodeAgent(
model=model,
tools=[final_answer, polite_guard, web_search, get_the_current_time_in_timezone ], ## add your tools here (don't remove final answer)
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()
''' |