@@ -55,30 +55,13 @@ async def prompt_to_choose_meal_tree(request_data: Payload) -> dict:
55
55
json_payload = request_data .payload
56
56
agent = Agent ()
57
57
agent .set_user_session (json_payload ["user_id" ], json_payload ["session_id" ])
58
- output = agent .prompt_to_choose_meal_tree (json_payload ["prompt" ], model_speed = json_payload ["model_speed" ])
58
+ output = agent .prompt_to_choose_meal_tree (json_payload ["prompt" ], model_speed = json_payload ["model_speed" ], assistant_category = "food" )
59
59
60
60
result = json .dumps ({"results" : list (map (splitter , output .replace ('"' , '' ).split (";" )))})
61
61
62
62
return JSONResponse (content = {"response" :json .loads (result )})
63
63
64
64
65
- @app .post ("/prompt-to-choose-meal-tree" , response_model = dict )
66
- async def prompt_to_choose_meal_tree (request_data : Payload ) -> dict :
67
- if CANNED_RESPONSES :
68
- with open ('fixtures/choose_meal_tree_response.json' , 'r' ) as f :
69
- json_data = json .load (f )
70
- stripped_string_dict = {"response" : json_data }
71
- return JSONResponse (content = stripped_string_dict )
72
-
73
-
74
- json_payload = request_data .payload
75
- agent = Agent ()
76
- agent .set_user_session (json_payload ["user_id" ], json_payload ["session_id" ])
77
- output = agent .prompt_to_choose_meal_tree (json_payload ["prompt" ], model_speed = json_payload ["model_speed" ])
78
-
79
- return JSONResponse (content = {"response" :output })
80
- from typing import Any , Generator
81
- from starlette .responses import StreamingResponse
82
65
@app .post ("/prompt-to-decompose-meal-tree-categories" , response_model = dict )
83
66
async def prompt_to_decompose_meal_tree_categories (request_data : Payload )-> dict :
84
67
json_payload = request_data .payload
@@ -107,6 +90,15 @@ async def prompt_to_update_meal_tree(request_data: Payload) -> dict:
107
90
print ("HERE IS THE OUTPUT" , output )
108
91
return JSONResponse (content = {"response" :output })
109
92
93
+ @app .post ("/fetch-user-summary" , response_model = dict )
94
+ async def fetch_user_summary (request_data : Payload ) -> dict :
95
+ json_payload = request_data .payload
96
+ agent = Agent ()
97
+ agent .set_user_session (json_payload ["user_id" ], json_payload ["session_id" ])
98
+ output = agent .fetch_user_summary ( model_speed = json_payload ["model_speed" ])
99
+ print ("HERE IS THE OUTPUT" , output )
100
+ return JSONResponse (content = {"response" :output })
101
+
110
102
111
103
@app .post ("/recipe-request" , response_model = dict )
112
104
async def recipe_request (request_data : Payload ) -> dict :
@@ -149,7 +141,7 @@ async def voice_input(request_data: Payload) -> dict:
149
141
agent = Agent ()
150
142
agent .set_user_session (json_payload ["user_id" ], json_payload ["session_id" ])
151
143
output = agent .voice_input (query = json_payload ["query" ], model_speed = json_payload ["model_speed" ])
152
- return JSONResponse (content = {"response" :json . loads ( output ) })
144
+ return JSONResponse (content = {"response" :output })
153
145
154
146
@app .get ("/health" )
155
147
def health_check ():
0 commit comments