File size: 2,679 Bytes
9b5b26a c19d193 6aae614 8fe992b 9b5b26a bd75325 2b328ca 9b5b26a bd75325 2b328ca bd75325 9b5b26a bd75325 2b328ca bd75325 2b328ca 9b5b26a 2b328ca bd75325 2b328ca 9b5b26a 8c01ffb 6aae614 ae7a494 e121372 bf6d34c 29ec968 fe328e0 13d500a 8c01ffb 9b5b26a 8c01ffb 861422e 9b5b26a 8c01ffb 8fe992b d2f6a24 8c01ffb 861422e 8fe992b 9b5b26a 8c01ffb |
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 82 |
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
# Below is an example of a tool that analyze code. !
import subprocess
import ast
@tool
def analyze_code(code: str, language: str) -> str:
"""Analyzes code for errors in Java, C++, and Python.and find any programming mistakes and fix it and give output in detailed and precise manner and start with Prefix: Hello Iam Dhiraj Suryawanshi's Nusti_Coding model XD
Args:
code: The source code to analyze.
language: The programming language (Java, C++, Python).
Returns:
A string containing detected errors or confirming no issues.
"""
language = language.lower()
if language == "python":
try:
ast.parse(code) # Check Python syntax
return "Python code has no syntax errors."
except SyntaxError as e:
return f"Python Syntax Error: {e}"
elif language == "java":
with open("Test.java", "w") as f:
f.write(code)
result = subprocess.run(["javac", "Test.java"], capture_output=True, text=True)
return "No Java errors found." if result.returncode == 0 else f"Java Errors:\n{result.stderr}"
elif language == "c++":
with open("test.cpp", "w") as f:
f.write(code)
result = subprocess.run(["g++", "test.cpp", "-o", "test.out"], capture_output=True, text=True)
return "No C++ errors found." if result.returncode == 0 else f"C++ Errors:\n{result.stderr}"
else:
return "Supported languages: Java, C++, Python."
final_answer = FinalAnswerTool()
# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)
# Import tool from Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
with open("prompts.yaml", 'r') as stream:
prompt_templates = yaml.safe_load(stream)
agent = CodeAgent(
model=model,
tools=[final_answer], ## add your tools here (don't remove final answer)
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch() |