community[patch]: Resolve parse_obj (#26262)

Update parse_obj
This commit is contained in:
Eugene Yurtsev
2024-09-10 15:56:34 -04:00
committed by GitHub
parent ed412d89b4
commit bb83f1b875

View File

@@ -416,7 +416,7 @@ class ChatKinetica(BaseChatModel):
# convert the prompt to messages # convert the prompt to messages
# request = SuggestRequest.model_validate(prompt_json) # pydantic v2 # request = SuggestRequest.model_validate(prompt_json) # pydantic v2
request = _KdtoSuggestRequest.parse_obj(prompt_json) request = _KdtoSuggestRequest.model_validate(prompt_json)
payload = request.payload payload = request.payload
dict_messages = [] dict_messages = []
@@ -448,7 +448,7 @@ class ChatKinetica(BaseChatModel):
data = response_json["data"] data = response_json["data"]
# response = CompletionResponse.model_validate(data) # pydantic v2 # response = CompletionResponse.model_validate(data) # pydantic v2
response = _KdtCompletionResponse.parse_obj(data) response = _KdtCompletionResponse.model_validate(data)
if response.status != "OK": if response.status != "OK":
raise ValueError("SQL Generation failed") raise ValueError("SQL Generation failed")
return response.data return response.data