recommendations working

plan/recommend.py

@@ -5,6 +5,33 @@ def recommend(payload):
 
     from langchain_core.pydantic_v1 import BaseModel, Field
 
+    class InputGoals(BaseModel):
+        """
+            The existing goals that the user has associated with this particular ambition.
+            You should pay close attention to this, and avoid repetition.
+            You might build on goals that the user has already resolved, or that have been accepted.
+            You should pay attention to goals that the user has rejected, and avoid recommending similar goals.
+        """
+
+        goal: str = Field(description="the goal itself in a clear format")
+        bucket: str = Field(description="""
+            The timescale within which the goal is currently scheduled to be completed. This will have one of three values:
+                - 'now' a goal that could be completed immediately with ease
+                - 'soon' a goal that requires a bit more planning or research, but that should be  achievable within the current academic year
+                - 'during' a goal that might take considerable research and effort to achieve, and may take longer than a single academic year.
+            In making your recommendations, you should consider how they fit in with these other goals
+                """)
+        status: str = Field(description="""
+                indicates the origin and current progress towards the goal. 
+                This will have one of three values, and these should guide your response: 
+                    - 'user_provided' the user has provided this goal themselves.
+                    - 'suggestion' you have suggested this goal previously, and the user has not yet accepted or rejected it.
+                    - 'accepted' you have suggested this goal previously, and the user has accepted it.
+                    - 'rejected' you have suggested this goal previously, and the user has rejected it.
+                    - 'resolved' the user has completed this goal.
+                    - 'removed' the user has removed this goal.
+            """)
+
     class UserInput(BaseModel):
         """The format of information that will be provided by the user"""
 
@@ -15,21 +42,28 @@ def recommend(payload):
             - '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. 
+            Whilst you can and should recommend small standalone courses, and applying for other full time or substantial courses after the current course is completed, you should NEVER suggest that the user should take on a full time course whilst they are already studying.
             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")
+        plan: List[InputGoals] = Field(description="The existing goals that the user has associated with this particular ambition.")
 
     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")
+        """
+            Your suggestion for a suggested goal. 
+            In forming this you should ALWAYS pay attention to the information provided in the user input, and the guidance provided in the JSON schema for it
+            Where possible, avoid recommending that the user 'researches' something, as this is a very vague goal
+            And avoid referring to the careers service or advisers.
+        """
+
+        goal: Optional[str] = Field(default=None, description="a 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)")
-        
+        guidance: Optional[str] = Field(default=None, description="provide any additional guidance that the user might need to achieve this goal")
+        #inputstring: Optional[str] = Field(default=None, description="the previous goals that have been provided by the user")
+        count_reject: Optional[int] = Field(default=0, description="the number of times that the user has rejected a goal")
 
     from typing import Optional
 
@@ -52,9 +86,7 @@ def recommend(payload):
 
                     {input_format}
 
-                    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.
-                    
+                    You should respond in the formal provided                    
                 """,
             ),
             (
@@ -64,6 +96,7 @@ def recommend(payload):
     )
 
     llm = ChatOpenAI(model="gpt-3.5-turbo-0125", temperature=0.5);
+    #llm=ChatOpenAI(model="gpt-4", temperature=0)
 
     runnable = prompt | llm.with_structured_output(schema=Goal)
 
@@ -71,6 +104,8 @@ def recommend(payload):
 
     response = runnable.invoke({"input":input, "input_format": inputformat})
 
+    #print(input);
+
     print(response.dict());
 
     return response.dict();