new input format

plan/recommend.py

@@ -1,14 +1,35 @@
 def recommend(payload):
 
     from typing import Optional, List
+    from langchain.output_parsers import PydanticOutputParser
 
     from langchain_core.pydantic_v1 import BaseModel, Field
 
+    class UserInput(BaseModel):
+        """The format of information that will be provided by the user"""
+
+        ambition: Optional[str] = Field(default=None, description="the ambition that the user is working on")
+        bucket: Optional[str] = Field(default=None, description="""
+            The timescale within which the goal should be able to be completed. This will have one of three values, and these should guide your response:
+            - 'now' you should provide a goal that could be completed easily and immediately
+            - 'soon' you should recommend a goal that might require a bit more research or planning, but that should still be achievable within the current academic year
+            - 'during' you should provide a goal that might take considerable research and effort to achieve, and may take longer than a single academic year.
+            Please be mindful that the user should always be able to complete all of their goals whilst they are on their current course. 
+            Part time additional effort is reasonable, but do not recommend things that will require a full time commitment. 
+        """)
+        plan: Optional[str] = Field(default=None, description="the user's current plan")
+
+    inputformat = PydanticOutputParser(pydantic_object=UserInput).get_format_instructions();
+    inputformat = inputformat.replace("The output should be formatted as a JSON instance that conforms to the JSON schema below.", "The user will provide their input as a JSON instance that conforms to the JSON schema below")
+    inputformat = inputformat.replace("Here is the output schema:", "Here is the input schema:")
+    
     class Goal(BaseModel):
         """Information about a suggested goal."""
 
         goal: Optional[str] = Field(default=None, description="a SMART target for the user - this should be as concise as possible")
         rationale: Optional[str] = Field(default=None, description="explain why you have chosen this goal")
+        inputstring: Optional[str] = Field(default=None, description="the JSON input provided by the user (for debugging)")
+        
 
     from typing import Optional
 
@@ -22,21 +43,23 @@ def recommend(payload):
             (
                 "system",
                 """
-                    You are an advice agent supporting students at a post-92 university in England - your advice should be tailored to the context.
 
-                    The user will provide an ambition and you will provide a single SMART target that will help them to reach that ambition.
+                    You are a life and career coach for students at Manchester Metropolitan University, a post-92 institution in England with large numbers of non-traditional students.
+
+                    Your function is to help students to make plans to help them make the most of their time at university, to form ambitions and do things that will set them on track to achieve them.                 
+                    
+                    The user will provide you with information as a JSON blob in the following format:
 
-                    They will also provide a timescale for the desired goal:
+                    {input_format}
 
-                    - Now: you should recommend a goal that can easily be accomplished immediately
-                    - Soon: you should recommend a goal that can be accomplished within the next 3 to 6 months, but that might require some research and planning
-                    - During: recommend a goal that can be completed before the student finishes their degree. This could require a significant amount of work and planning.
-                            
-                    Here is the user's ambition: {ambition}
-                    They would like a goal that can be accomplished: {bucket}.
+                    You should respond with a concise goal that the user can work towards, and a rationale for why you have chosen this goal.
+                    Your goal should be SMART: Specific, Measurable, Achievable, Relevant, and Time-bound.
                     
                 """,
             ),
+            (
+                "user","{input}"
+            )
         ]
     )
 
@@ -44,15 +67,10 @@ def recommend(payload):
 
     runnable = prompt | llm.with_structured_output(schema=Goal)
 
-    input = {
-        "ambition": payload.get("ambition") or "",
-        "bucket": payload.get("bucket") or "",
-    }
-    
-    print(input);
+    input = payload.get("input");
+
+    response = runnable.invoke({"input":input, "input_format": inputformat})
 
-    response = runnable.invoke(input)
-    
     print(response.dict());
 
     return response.dict();