agent.py

# import os
# from PIL import Image
# from smolagents import CodeAgent, HfApiModel, InferenceClientModel

# import tools.tools as tls

# # --- Basic Agent Definition ---
# # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------

# class BasicAgent:
#     def __init__(self):
#         print("BasicAgent initialized.")
#     def __call__(self, question: str) -> str:
#         model = HfApiModel(model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud", provider="together", use_auth_token=True)

#         agent = CodeAgent(
#             tools=[tls.search_tool, tls.calculate_cargo_travel_time],
#             model=InferenceClientModel(),
#             additional_authorized_imports=["pandas"],
#             max_steps=20,
#         )

#         fixed_answer = agent.run(question)

#         print(f"Agent received question (first 50 chars): {question[:50]}...")
#         # fixed_answer = "This is a default answer."
#         print(f"Agent returning fixed answer: {fixed_answer}")
#         return str(fixed_answer)