File size: 2,938 Bytes
9b5b26a c19d193 6aae614 8fe992b 9b5b26a bd75325 2b328ca 6d27ba2 9b5b26a bd75325 6d27ba2 9b5b26a bd75325 2b328ca 6d27ba2 bd75325 2b328ca 9b5b26a 2b328ca 6d27ba2 2b328ca 6d27ba2 2b328ca 6d27ba2 2b328ca 6d27ba2 2b328ca 6d27ba2 bd75325 2b328ca 6d27ba2 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 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
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
import subprocess
import ast
@tool
def analyze_code(code: str, language: str) -> str:
"""Analyzes code for errors in Java, C++, and Python, finds programming mistakes,
fixes them, and provides detailed and precise output.
Prefix: Hello, I am 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()
prefix = "Hello, I am Dhiraj Suryawanshi's Nusti_Coding model XD\n"
if language == "python":
try:
ast.parse(code) # Check Python syntax
return prefix + "✅ Python code has no syntax errors."
except SyntaxError as e:
return prefix + 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)
if result.returncode == 0:
return prefix + "✅ No Java errors found."
else:
return prefix + 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)
if result.returncode == 0:
return prefix + "✅ No C++ errors found."
else:
return prefix + f"❌ C++ Errors:\n{result.stderr}"
else:
return prefix + "⚠️ 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() |