idZddlZddlZddlZddlZej eZdZdZ dZ dZ dZ de fd Zd ejd ee deeefd Zd e deefdZd ee deeefdZy)u embedding_service.py OpenAI Embedding API를 사용하는 임베딩 서비스. - 단일 텍스트 임베딩: get_embedding() - 배치 임베딩: get_embeddings_batch() Nztext-embedding-3-smallidreturnc^tjjd}|s td|S)u 환경변수에서 OpenAI API 키를 읽어 반환한다. Raises: ValueError: OPENAI_API_KEY 환경변수가 설정되지 않은 경우. OPENAI_API_KEYuOPENAI_API_KEY 환경변수가 설정되지 않았습니다. 올바른 API 키를 OPENAI_API_KEY 환경변수에 설정하세요.)osenvironget ValueErrorapi_keys G/home/jay/workspace/.worktrees/task-2116-dev1/libs/embedding_service.py _get_api_keyrs4jjnn-.G  S   NclienttextscPd}ttD]O} |jj|tt }|j Dcgc]}|j}}|cStj!dt|J|cc}w#tj$rL}|}td|zz}tjd|dzt|tj|Yd}~d}~wtj$r,}|}tjd|dzt|Yd}~d}~wwxYw)u OpenAI embeddings.create를 호출하며 에러 시 재시도한다. 재시도 정책: - openai.RateLimitError (HTTP 429): 지수 백오프(1, 2, 4초) 후 재시도 - openai.APIError: 최대 3회(초기 시도 포함) 재시도 - 3회 모두 실패 시 원래 예외를 raise Args: client: OpenAI 클라이언트 인스턴스. texts: 임베딩할 텍스트 목록. Returns: 임베딩 벡터 목록. 각 벡터는 1536차원 float 리스트. Raises: openai.APIError: 최대 재시도 횟수 초과 시. openai.RateLimitError: 최대 재시도 횟수 초과 시. N)inputmodel dimensionsuFRate limit 에러 발생 (시도 %d/%d). %d초 후 재시도합니다.ru$API 에러 발생 (시도 %d/%d): %su@최대 재시도 횟수(%d회) 초과. 에러를 raise합니다.)range _MAX_RETRIES embeddingscreate_EMBEDDING_MODEL_EMBEDDING_DIMENSIONSdata embeddingopenaiRateLimitError _BACKOFF_BASEloggerwarningtimesleepAPIErrorerror) rrlast_exceptionattemptresponseitemvectorsexcbackoffs r_call_embeddings_with_retryr1's.(,N& ((//&00H FN]])ST$..)SG)SN> LLSUab  %% % 5*T$$  N#q'z2G NNX!    JJw    N NN6!     s64B  BB B  D%AC##D%9!D  D%textct}tj|}tj d|ddt ||g}|dS)uM 단일 텍스트의 임베딩 벡터를 반환한다. Args: text: 임베딩할 텍스트 문자열. Returns: 1536차원 float 리스트. Raises: ValueError: OPENAI_API_KEY 환경변수가 설정되지 않은 경우. openai.APIError: API 호출이 3회 모두 실패한 경우. r u%단일 텍스트 임베딩 요청: %rN2r)rr!OpenAIr$debugr1)r2rrr.s r get_embeddingr7dsFnG ]]7 +F LL8$s)D)&4&9G 1:rc 4|sgSt}tj|}g}t|}t d|t D]R}|||t z}t jd|dz|t|z|t||}|j|T|S)u 여러 텍스트의 임베딩 벡터를 배치로 반환한다. 100개를 초과하는 경우 100개씩 분할하여 여러 번 API를 호출한다. Args: texts: 임베딩할 텍스트 문자열 목록. Returns: 각 텍스트에 대응하는 1536차원 벡터 목록. 입력이 빈 리스트인 경우 빈 리스트를 반환한다. Raises: ValueError: OPENAI_API_KEY 환경변수가 설정되지 않은 경우. openai.APIError: API 호출이 3회 모두 실패한 경우. r ru*배치 임베딩 요청: %d~%d / 총 %d개r) rr!r5lenr _BATCH_SIZEr$r6r1extend)rrr all_vectorstotal batch_startbatch batch_vectorss rget_embeddings_batchrAzs"  nG ]]7 +F%'K JEQ{3 * kK+$=> 8 !O #e* $   4FEB =) * r)__doc__loggingr r&r! getLogger__name__r$rrr:rr#strrr5listfloatr1r7rArrrJs   8 $+    c : MM: 9: $u+:zU ,%S %d4;.?%r