python_coding_agent.py

import warnings
warnings.filterwarnings("ignore")
from typing import Callable
from helper import load_env, get_openai_api_key
from openai import OpenAI
from llm import LLM
from typing import TypedDict
import sys
from io import StringIO
from tools import execute_code_schema
from execute_tool import execute_tool
from Execution import execute_code
import tools
client = OpenAI(
    base_url="https://openrouter.ai/api/v1",
    api_key=get_openai_api_key(),
)

messages = [{
    "role": "user",
    "content": "Hello, I am a Python coding agent. How can I assist you today?"
}]

system_prompt = (
    "You are a helpful Python coding agent. You will be given a task in Python. "
    "Respond with code only, no explanations."
)

response = LLM(
    client,
    messages,
    model="mistralai/mistral-nemo",
    system_prompt=system_prompt
).get_response()

print(response)

result = execute_code(response)
print("Results:", result["results"])
print("Errors:", result["errors"])