Update app.py

app.py

@@ -23,12 +23,13 @@ from utils import parse_action, parse_file_content, read_python_module_structure
 from datetime import datetime
 
 def format_prompt(message, history):
-  prompt = "<s>"
-  for user_prompt, bot_response in history:
-    prompt += f"[INST] {user_prompt} [/INST]"
-    prompt += f" {bot_response}</s> "
-  prompt += f"[INST] {message} [/INST]"
-  return prompt
+    """Formats the prompt by concatenating user prompts and bot responses."""
+    prompt = "<s>"
+    for user_prompt, bot_response in history:
+        prompt += f"[INST] {user_prompt} [/INST]"
+        prompt += f" {bot_response}</s> "
+    prompt += f"[INST] {message} [/INST]"
+    return prompt
 
 def run_gpt(
     prompt_template,
@@ -37,6 +38,7 @@ def run_gpt(
     purpose,
     **prompt_kwargs,
 ):
+    """Runs the GPT model to generate a response."""
     seed = random.randint(1, 1111111111111111)
     print(seed)
     generate_kwargs = dict(
@@ -53,7 +55,7 @@ def run_gpt(
         purpose=purpose,
         safe_search=safe_search,
     ) + prompt_template.format(**prompt_kwargs)
-    
+
     if True:
         print(LOG_PROMPT.format(content))
 
@@ -67,6 +69,7 @@ def run_gpt(
 
 
 def compress_history(purpose, task, history, directory):
+    """Compresses the history using the GPT model."""
     resp = run_gpt(
         COMPRESS_HISTORY_PROMPT,
         stop_tokens=["observation:", "task:", "action:", "thought:"],
@@ -78,6 +81,7 @@ def compress_history(purpose, task, history, directory):
     return resp
 
 def call_search(purpose, task, history, directory, action_input):
+    """Calls the search function."""
     print("CALLING SEARCH")
     try:
 
@@ -98,6 +102,7 @@ def call_search(purpose, task, history, directory, action_input):
     return "MAIN", None, history, task
 
 def call_main(purpose, task, history, directory, action_input):
+    """Calls the main function."""
     resp = run_gpt(
         ACTION_PROMPT,
         stop_tokens=["observation:", "task:", "action:","thought:"],
@@ -134,6 +139,7 @@ def call_main(purpose, task, history, directory, action_input):
 
 
 def call_set_task(purpose, task, history, directory, action_input):
+    """Sets the task."""
     task = "COMPLETE"
     resp = run_gpt(
         TASK_PROMPT,
@@ -147,6 +153,7 @@ def call_set_task(purpose, task, history, directory, action_input):
     return "MAIN", None, history, task
 
 def end_fn(purpose, task, history, directory, action_input):
+    """Ends the function."""
     task = "COMPLETE"
     return "COMPLETE", "COMPLETE", history, task
 
@@ -159,6 +166,7 @@ NAME_TO_FUNC = {
 }
 
 def run_action(purpose, task, history, directory, action_name, action_input):
+    """Runs the action."""
     print(f'action_name::{action_name}')
     try:
         if "RESPONSE" in action_name or "COMPLETE" in action_name:
@@ -185,7 +193,7 @@ def run_action(purpose, task, history, directory, action_name, action_input):
         return "MAIN", None, history, task
 
 def run(purpose,history):
-
+    """Runs the main loop."""
     #print(purpose)
     #print(hist)
     task = "COMPLETE"
@@ -229,6 +237,7 @@ agents =[
 def generate(
         prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
+    """Generates the output."""
     seed = random.randint(1,1111111111111111)
 
     agent=prompts.WEB_DEV
@@ -356,6 +365,7 @@ examples = [
 ]
 
 def launch_interface():
+    """Launches the interface."""
     gr.ChatInterface(
         fn=run,
         chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, layout="panel"),