Spaces:
Sleeping
Sleeping
File size: 1,993 Bytes
959296e c92c3bf d93dbdd 9b5b26a 0f61d7b 2764a1c 1c61c68 4c2a00d d4578f1 4db54a2 e55e264 67c8f73 7686479 ed36d14 95de6d7 2945906 c14a3d3 9b5b26a 8f16109 95c49b7 826fb71 65b9821 6dc9c22 06e3381 95c49b7 a22a330 0e921e3 a22a330 0e921e3 06e3381 6387f36 7155a63 0f61d7b 4999ab5 a22a330 4999ab5 a22a330 4999ab5 351e87d 4999ab5 b356cef 4999ab5 b356cef 1c61c68 |
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 79 80 81 |
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 ContentGradio import ContentAgentUI
# Prechecks
# Get the URI for the endpoint
endpoint_uri = load_huggingface_model()
# Test the endpoint
status_info = check_public_endpoint(endpoint_uri)
print(status_info["status"])
print(status_info["status_code"])
if status_info["status"] is False:
interface = create_failed_gradio_ui(status_info)
# Launch the UI
interface.launch(show_error=True)
else:
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id= endpoint_uri,
custom_role_conversions=None,
)
# Load prompts
combined_prompts = load_prompts()
combined_tools = load_tools() ## from the tools folder (don't remove final answer)
# Try a direct test of your endpoint
test_prompt = "Hello, how are you?"
response = requests.post(
endpoint_uri,
json={"inputs": test_prompt},
headers={"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
)
print(response.json())
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"]
)
# Create an instance of the ContentAgentUI class
ui = ContentAgentUI()
# Pass through the agent
ui.pass_through_agent(agent)
|