Spaces:
Sleeping
Sleeping
File size: 1,471 Bytes
959296e c92c3bf d93dbdd 9b5b26a 1c61c68 4c2a00d d4578f1 4db54a2 56df9e5 67c8f73 65b9821 ed36d14 95de6d7 9b5b26a d4578f1 3335de4 65b9821 67c8f73 65b9821 1cde8a3 aa945ae 1dfe865 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 |
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import sys # Add this import to fix the error
import datetime
import requests
import os
import pytz # Had to give it permission in Code agent
from tools.tools import tools
from check_endpoint import is_huggingface_endpoint
from agents.prompts import load_prompts
from Gradio_UI import GradioUI
my_id = os.getenv("QWEN_URI")
# Test the endpoint
if is_huggingface_endpoint(my_id):
print("This is a Hugging Face Inference Endpoint.")
else:
print("This is NOT a Hugging Face Inference Endpoint.")
sys.exit(1) # Stop execution if the endpoint is not valid
'''
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id= my_id,
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()
''' |