Merge pull request #76 from studio-recoding/refactor/delete

refactor: delete 기능 리팩토링
studio-recoding · Jul 8, 2024 · 679caf4 · 679caf4
2 parents 9eeace2 + fe54ab5
commit 679caf4
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/app/database/chroma_db.py b/app/database/chroma_db.py
@@ -75,6 +75,7 @@ async def add_db_data(schedule_data: AddScheduleDTO):
             "member": schedule_data.member_id,
             "category": schedule_data.category,
             "category_id": schedule_data.schedule_id,
+            "category_color": schedule_data.category_color,
             "location": schedule_data.location,
             "person": schedule_data.person
         }]

diff --git a/app/dto/db_dto.py b/app/dto/db_dto.py
@@ -24,6 +24,7 @@ class UpdateScheduleDTO(BaseModel):
     member_id: int
     category: str
     category_id: int
+    category_color: str
     location: str
     person: str
 

diff --git a/app/prompt/openai_prompt.py b/app/prompt/openai_prompt.py
@@ -194,6 +194,7 @@ class Template:
             
             1. Respond kindly to the user's input. YOU MUST USE {output_language} TO RESPOND TO THE USER INPUT.
             2. You will be given a list of potential candidates from the database for events that the user may want to delete, and you must organize those events that the user has indicated they want to delete into a list. Organize the events the user wants to delete into a json format to make a delete api call in a database. Each event should be represented as a separate json object within a list. Each json object will have keys for info, location, person, start_time, end_time, and category. The category should include the name, id, and color. 
+            - id: Find the id of the schedule in the 'ids'.
             - info: The document data of the schedule.
             - location: Include the venue or place where the event was scheduled to occur.
             - person: List any specific individuals involved in the event.
@@ -214,6 +215,7 @@ class Template:
             <separate>
             [
                 {{
+                    "id": 29
                     "info": "개발 공부하기",
                     "location": "공대",
                     "person": "",
@@ -226,6 +228,7 @@ class Template:
                     }}
                 }},
                 {{
+                    "id": 7
                     "info": "한강 놀러가기",
                     "location": "한강",
                     "person": "혜승",