Update backend_agent/api_generator.py

backend_agent/api_generator.py

@@ -8,13 +8,14 @@ model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
 
 def generate_backend_code_llm(task_name):
     """
-    Generate Python REST/GraphQL backend code for a task
+    Generates Python backend code (REST/GraphQL) for a given task using LLM.
     """
     prompt = f"""
-You are an expert backend developer. Generate Python code for a REST or GraphQL API for this backend task:
+You are an expert backend developer. Generate Python code for REST or GraphQL APIs
+for the following task:
 Task: {task_name}
 
-Include routes/resolvers, input validation placeholders, and comments.
+Include routes/resolvers, input validation, and placeholders for business logic.
 """
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(**inputs, max_new_tokens=350)
@@ -22,5 +23,4 @@ Include routes/resolvers, input validation placeholders, and comments.
 
     if "Task:" in code:
         code = code.split("Task:")[-1].strip()
-
     return code