Merge pull request #72 from studio-recoding/feat/case2-multiple-sched…

…ules hotfix: 에러 수정
studio-recoding · Jun 26, 2024 · cec5256 · cec5256
2 parents 9119f32 + 838f6e4
commit cec5256
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 9 deletions.
diff --git a/app/prompt/openai_config.ini b/app/prompt/openai_config.ini
@@ -1,7 +1,7 @@
 [NESS_NORMAL]
 TEMPERATURE     = 0.5
 MAX_TOKENS      = 1500
-MODEL_NAME      = gpt-4-turbo
+MODEL_NAME      = gpt-4
 
 [NESS_CASE]
 TEMPERATURE     = 0

diff --git a/app/prompt/openai_prompt.py b/app/prompt/openai_prompt.py
@@ -136,30 +136,30 @@ class Template:
             Shall I add your meeting with Dr. Smith at her office on March 3rd from 10am to 11am and your dinner with John at the Italian restaurant on March 4th at 7pm to your schedule?
             <separate>
             [
-              {
+              {{
                 "info": "meeting with Dr. Smith",
                 "location": "Dr. Smith's office",
                 "person": "Dr. Smith",
                 "start_time": "2023-03-03T10:00:00+09:00",
                 "end_time": "2023-03-03T11:00:00+09:00",
-                "category": {
+                "category": {{
                   "name": "Work",
                   "id": 1,
                   "color": "#FF0000"
-                }
-              },
-              {
+                }}
+              }},
+              {{
                 "info": "dinner with John",
                 "location": "Italian restaurant",
                 "person": "John",
                 "start_time": "2023-03-04T19:00:00+09:00",
                 "end_time": null,  // Assuming end time is not specified
-                "category": {
+                "category": {{
                   "name": "Personal",
                   "id": 2,
                   "color": "#00FF00"
-                }
-              }
+                }}
+              }}
             ]
             
             User input: {question}

diff --git a/app/routers/chat.py b/app/routers/chat.py
@@ -151,6 +151,7 @@ async def get_langchain_schedule(data: PromptRequest, chat_type_prompt):
         return response
 
     except Exception as e:
+        print(e)
         raise HTTPException(status_code=500, detail=str(e))
 
 # case 3 : rag