Update prompts

main.py

@@ -25,6 +25,70 @@ SysPromptMd = "You are an expert AI who can create a structured report using inf
 SysPromptMdOffline = "You are an expert AI who can create a structured report using your knowledge on user request.The report should be in markdown format consists of markdown tables/lists/paragraphs as needed, structured into subtopics. Do not add any additional comments."
 
 
+prompt_topics = """
+You are an expert data analyst. You have been providing data analysis and structuring services for over 15 years. You specialize in creating detailed and comprehensive lists of subtopics for various fields.
+
+Your task is to create a list of essential subtopics along with their descriptions based on the given USER_QUERY. The objective is to generate a detailed and precise list of subtopics that can help in understanding the main topic thoroughly.
+
+Follow these steps to complete the task:
+
+1. Identify and list 2 to {num_topics} essential subtopics related to {user_input}.
+2. Provide a detailed description for each subtopic explaining its significance and relevance to the main topic.
+3. Format your response as a valid Python list of lists, where each sub-list contains the subtopic and its description.
+
+Make sure your response is well-organized and provides comprehensive details for each subtopic.
+
+Take a deep breath and work on this problem step-by-step.
+
+output format
+[
+    ["Subtask Title 1", "Detailed description of subtask 1."],
+    ["Subtask Title 2", "Detailed description of subtask 2."],
+    ...
+]
+YOUR OUTPUT SHOULD CONSIST ONLY A VALID PYTHON LIST, DO NOT ADD ADDITIONAL COMMENTS
+"""
+
+prompt_subtopics = """You are a professional task manager and prompt engineer. You have been helping teams and individuals decompose complex tasks into actionable subtasks for over 20 years. Your expertise lies in breaking down intricate tasks into clear, manageable steps and ensuring that all relevant aspects are covered while excluding any specified topics.**
+
+Objective: Help create 2 to {num_topics} subtasks for an LLM to perform the specified task in the context of the given user query. Ensure that the generated subtasks are precise, actionable, and detailed. Exclude the specified topics from the subtasks.
+
+**Steps to complete the task:**
+
+1. **Understand the Main Task and User Query:**
+   - Read the main task and user query carefully to grasp the core objectives and context.
+   - Identify any specific requirements or constraints mentioned in the query.
+
+2. **Identify Key Components:**
+   - Break down the main task into its fundamental components.
+   - Ensure each component is essential to the overall goal and can be clearly defined.
+
+3. **Create Detailed Subtasks:**
+   - For each key component, create 2 to 5 detailed subtasks.
+   - Ensure each subtask is actionable and provides clear instructions on what needs to be done.
+   - Maintain logical order and ensure that the completion of each subtask contributes to the overall objective.
+
+4. **Exclude Specified Topics:**
+   - Review the list of topics to be excluded.
+   - Ensure that none of the generated subtasks include these topics.
+
+5. **Format the Response:**
+   - Present the subtasks in a structured Python list of lists format.
+   - Each sub-list should contain the subtask title and its detailed description.
+
+Output Format:
+[
+    ["Subtask Title 1", "Detailed description of subtask 1."],
+    ["Subtask Title 2", "Detailed description of subtask 2."],
+    ...
+]
+
+MAIN TASK: {main_task}
+USER QUERY:{user_input}
+TOPICS TO BE EXCLUDED:{excluded_topics}
+YOUR OUTPUT SHOULD CONSIST ONLY A VALID PYTHON LIST, DO NOT ADD ADDITIONAL COMMENTS
+"""
+
 def together_response(message, model = "meta-llama/Llama-3-8b-chat-hf", SysPrompt = SysPromptDefault, temperature=0.2):
     client = OpenAI(
         api_key=TOGETHER_API_KEY, 
@@ -61,11 +125,8 @@ def json_from_text(text):
         return loads(fix_json.fix(json_out).line)
     
 def generate_topics(user_input, num_topics, previous_queries):
-    previous_context = " -> ".join(previous_queries)
-    prompt = f"""create a list of {num_topics} subtopics along with descriptions to follow for conducting {user_input} in the context of {previous_context}, RETURN A VALID PYTHON LIST"""\
-                +""" Respond in the following format: 
-        [["Subtopic","Description"],["Subtopic","Description"]]"""
-    response_topics = together_response(prompt, model="meta-llama/Llama-3-8b-chat-hf", SysPrompt=SysPromptList)
+    prompt = prompt_topics.format(user_input=user_input, num_topics=num_topics)
+    response_topics = together_response(prompt, model="meta-llama/Llama-3-70b-chat-hf", SysPrompt=SysPromptList, temperature=1)
     subtopics = json_from_text(response_topics)
     return subtopics
 
@@ -95,7 +156,13 @@ app.add_middleware(
 class TopicInput(BaseModel):
     user_input: str = Query(default="market research", description="input query to generate subtopics")
     num_topics: int = Query(default=5, description="Number of subtopics to generate (default: 5)")
-    previous_queries: list[str] = Query(default=[], description="List of previous queries for context")
+    previous_queries: list[str] = Query(default=[], description="Deprecated: Use /generate_subtopics instead for subtopics")
+
+class SubTopicInput(BaseModel):
+    main_task: str = Query(default="detailed market research", description="Main task to be completed")
+    user_input: str = Query(default="I want to start a business in retail", description="input query to generate subtopics")
+    num_topics: int = Query(default=3, description="Number of max subtopics to generate (default: 3)")
+    excluded_topics: list[str] = Query(default=[], description="List all other main tasks to exclude")
 
 class imageInput(BaseModel):
     user_input: str = Query(default="market research", description="input query to generate subtopics")
@@ -118,6 +185,11 @@ async def create_topics(input: TopicInput):
     topics = generate_topics(input.user_input, input.num_topics, input.previous_queries)
     return {"topics": topics}
 
+@app.post("/generate_subtopics")
+async def create_subtopics(input: SubTopicInput):
+    topics = generate_subtopics(input.main_task, input.user_input, input.num_topics, input.excluded_topics)
+    return {"subtopics": topics}
+
 @app.post("/generate_report")
 async def create_report(input: ReportInput):
     report = generate_report(input.topic, input.description)