mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-27 22:37:46 +00:00
@@ -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
|
||||||
|
Reference in New Issue
Block a user