Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 from typing import List, Dict, Tuple, Any
 from pgvector.psycopg2 import register_vector
 import numpy as np
-from datetime import datetime, timezone
+from datetime import datetime, timezone, timedelta
 
 # 가중치 및 임계값 설정
 DEFAULT_FULL_WEIGHT = 0.2
@@ -43,9 +43,8 @@ def get_embedding(text: str) -> List[float]:
     try:
         response = client.embeddings.create(
             input=text,
-            model="text-embedding-3-small",
-            encoding_format="float",
-            dimensions=1536
+            model="text-embedding-ada-002",
+            encoding_format="float"
         )
         # 명시적으로 float32로 변환하여 Java의 float[]와 호환되게 함
         return np.array(response.data[0].embedding, dtype=np.float32).tolist()
@@ -167,7 +166,7 @@ def search_similar_chat(query: str, max_results: int = 100) -> List[Dict]:
                         "topic": get_text_value(metadata, "topic")
                     }
                     
-                    # 시간 필드 변환 없이 그대로 사용 (이미 KST로 저장되어 있음)
+                    # 시간 필드 변환 없이 그대로 사용
                     if "startTime" in metadata and metadata["startTime"] is not None:
                         result["startTime"] = metadata["startTime"]
                     
@@ -180,7 +179,7 @@ def search_similar_chat(query: str, max_results: int = 100) -> List[Dict]:
                     print(f"문제가 발생한 메타데이터: {metadata_json[:200]}...")
                     continue
             
-            # 임계값 필터링 (자바 코드와 동일하게 구현)
+            # 임계값 필터링
             filtered_results = [r for r in results if r["similarityScore"] >= threshold]
             
             if len(filtered_results) > 0:
@@ -230,27 +229,7 @@ def search_similar_chat_by_date(
     print(f"다중 임베딩 날짜 검색 시작: 쿼리='{query}', 시작일={start_date}, 종료일={end_date}, 최대 결과={limit}")
     
     try:
-        # 날짜 필터 파라미터 생성 (자바 코드와 동일하게 구현)
-        start_timestamp = None
-        end_timestamp = None
-        
-        if start_date and start_date.strip():
-            try:
-                # 자바에서는 LocalDateTime.parse() 사용하므로 동일하게 구현
-                start_datetime = datetime.strptime(start_date + "T00:00:00", '%Y-%m-%dT%H:%M:%S')
-                start_timestamp = int(start_datetime.timestamp() * 1000)  # 밀리초 단위로 변환
-            except ValueError as e:
-                print(f"시작 날짜 형식 오류: {str(e)}")
-                return []
-        
-        if end_date and end_date.strip():
-            try:
-                # 자바에서는 LocalDateTime.parse() 사용하므로 동일하게 구현
-                end_datetime = datetime.strptime(end_date + "T23:59:59", '%Y-%m-%dT%H:%M:%S')
-                end_timestamp = int(end_datetime.timestamp() * 1000)  # 밀리초 단위로 변환
-            except ValueError as e:
-                print(f"종료 날짜 형식 오류: {str(e)}")
-                return []
+        # 날짜 필터 생성
         
         # 쿼리 임베딩 생성
         query_embedding = get_embedding(query)
@@ -280,12 +259,16 @@ def search_similar_chat_by_date(
                 OR agent_embedding IS NOT NULL
         """ % (query_vector, full_w, query_vector, topic_w, query_vector, customer_w, query_vector, agent_w)
         
-        # 날짜 필터 추가 (자바 코드와 동일하게 구현)
-        if start_timestamp is not None:
-            sql += f" AND (metadata->>'startTime')::bigint >= {start_timestamp}"
+        # 날짜 필터 추가
+        if start_date and start_date.strip():
+            # 시작 시간 추가하여 ISO 형식으로 비교
+            iso_start_date = start_date + "T00:00:00"
+            sql += f" AND metadata->>'startTime' >= '{iso_start_date}'"
         
-        if end_timestamp is not None:
-            sql += f" AND (metadata->>'startTime')::bigint <= {end_timestamp}"
+        if end_date and end_date.strip():
+            # 종료 시간 추가하여 ISO 형식으로 비교
+            iso_end_date = end_date + "T23:59:59"
+            sql += f" AND metadata->>'startTime' <= '{iso_end_date}'"
         
         sql += """
             )
@@ -300,7 +283,7 @@ def search_similar_chat_by_date(
         """
         
         with conn.cursor() as cur:
-            print(f"날짜 검색 쿼리 실행: 시작일={start_date}({start_timestamp}), 종료일={end_date}({end_timestamp})")
+            print(f"날짜 검색 쿼리 실행: 시작일={start_date}, 종료일={end_date}")
             # 여기서는 limit를 파라미터로 전달
             cur.execute(sql, (limit,))
             rows = cur.fetchall()