o i@s@UddlZddlmZmZmZmZddlmZmZm Z m Z ddl m Z Gddde Z Gd d d e Zed d d eddddde de dddddddgdZeededdddee ddde ddde dd de d!d"de d#d$dgd$d%d&ddgdd d d eddddd'Zeed(<Gd)d*d*ejZed+kredSdS),N)*convert_to_parsed_chat_completion_responseParsedChatCompletionResponseParsedChatCompletionChoiceParsedAssistantMessage)ChatCompletionResponse UsageInfoChatCompletionChoiceAssistantMessage) BaseModelc@seZdZUeed<eed<dS) Explanation explanationoutputN)__name__ __module__ __qualname__str__annotations__rrl/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai/extra/tests/test_struct_chat.pyr s  r c@s"eZdZUeeed<eed<dS)MathDemonstrationsteps final_answerN)rrrlistr rrrrrrrs  r c0271b2098954c6094231703875ca0bczchat.completionzmistral-large-latestKi') prompt_tokenscompletion_tokens total_tokensiFgaM{ "final_answer": "x = -4", "steps": [ { "explanation": "Start with the given equation.", "output": "8x + 7 = -23" }, { "explanation": "Subtract 7 from both sides to isolate the term with x.", "output": "8x = -23 - 7" }, { "explanation": "Simplify the right side of the equation.", "output": "8x = -30" }, { "explanation": "Divide both sides by 8 to solve for x.", "output": "x = -30 / 8" }, { "explanation": "Simplify the fraction to get the final answer.", "output": "x = -4" } ] }F assistant)content tool_callsprefixrolestop)indexmessage finish_reason)idobjectmodelusagecreatedchoiceszStart with the given equation.z 8x + 7 = -23)rrz6Subtract 7 from both sides to isolate the term with x.z 8x = -23 - 7z(Simplify the right side of the equation.z8x = -30z&Divide both sides by 8 to solve for x.z x = -30 / 8z.Simplify the fraction to get the final answer.zx = -4)rr)r"r#r$r%parsed)r/r.r*r,r+r-expected_responsec@seZdZddZdS))TestConvertToParsedChatCompletionResponsecCsttt}||tdS)N)rmock_cc_responser assertEqualr1)selfrrrr/test_convert_to_parsed_chat_completion_response_szYTestConvertToParsedChatCompletionResponse.test_convert_to_parsed_chat_completion_responseN)rrrr6rrrrr2^s r2__main__)unittest struct_chatrrrrmodelsrr r r pydanticr r rr3r1rTestCaser2rmainrrrrs   % .