File tree Expand file tree Collapse file tree
apps/models_provider/impl/siliconCloud_model_provider/model Expand file tree Collapse file tree Original file line number Diff line number Diff line change 77 @desc:
88"""
99from typing import Dict , List
10-
10+ from common . utils . logger import maxkb_logger
1111import requests
1212
1313from models_provider .base_model_provider import MaxKBBaseModel
@@ -51,10 +51,13 @@ def embed_query(self, text: str) -> list:
5151
5252 response = requests .post (self .base_url + '/embeddings' , json = payload , headers = headers )
5353 data = response .json ()
54- if data ['data' ] is None or 'code' in data :
55- raise ValueError (f"Embedding API returned no data: { data } " )
56- # 假设返回结构中有 'data[0].embedding'
57- return data ["data" ][0 ]["embedding" ]
54+ if isinstance (data , dict ):
55+ if data ['data' ] is None or 'code' in data :
56+ raise ValueError (f"Embedding API returned no data: { data } " )
57+ # 假设返回结构中有 'data[0].embedding'
58+ return data ["data" ][0 ]["embedding" ]
59+ else :
60+ maxkb_logger .error (f"Unexpected response from Embedding API: { data } " )
5861
5962 def embed_documents (self , texts : list ) -> list :
6063 return [self .embed_query (text ) for text in texts ]
You can’t perform that action at this time.
0 commit comments