Merge pull request #36 from studio-recoding/feat/cicd

[refactor] metadata 추가
studio-recoding · Mar 24, 2024 · 69396f9 · 69396f9
2 parents 842fc01 + 2c6d620
commit 69396f9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/app/database/chroma_db.py b/app/database/chroma_db.py
@@ -50,7 +50,7 @@ async def add_db_data(schedule_data: AddScheduleDTO):
     schedules.add(
         documents=[schedule_data.data],
         ids=[str(schedule_data.schedule_id)],
-        metadatas=[{"datetime": schedule_data.schedule_datetime, "member": schedule_data.member_id, "category": schedule_data.category, "location": schedule_data.location, "person": schedule_data.person}]
+        metadatas=[{"datetime_start": schedule_data.schedule_datetime_start, "datetime_end": schedule_data.schedule_datetime_end, "member": schedule_data.member_id, "category": schedule_data.category, "location": schedule_data.location, "person": schedule_data.person}]
     )
     return True
 

diff --git a/app/dto/db_dto.py b/app/dto/db_dto.py
@@ -3,7 +3,8 @@
 
 class AddScheduleDTO(BaseModel):
     data: str
-    schedule_datetime: str  # datetime.datetime 타입을 명시적으로 사용합니다.
+    schedule_datetime_start: str  # datetime.datetime 타입을 명시적으로 사용합니다.
+    schedule_datetime_end: str
     schedule_id: int
     member_id: int
     category: str