3fiU$<ddlZddlmZddlmZmZmZmZmZm Z m Z m Z m Z m Z ddlmZddlZddlmZmZmZmZddlmZddlmZddlmZdd lmZdd lmZm Z m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'm(Z(dd l)m*Z*ddl+m,Z,m-Z-ddl.m/Z/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:m;Z;ejxejzddej|e?Z@dedeAfdZBdee deeeCeffdZDdejdee"fdZFedd d!"Gd#d$eZGy)%N) itemgetter) AnyCallableDictListLiteralOptionalSequenceTypeUnioncast)uuid4) AIMessageChatGeneration ChatResult HumanMessage) deprecated)CallbackManagerForLLMRun)LanguageModelInput) BaseChatModel)AIMessageChunk BaseMessage SystemMessageToolCall ToolMessage) tool_call)JsonOutputParserPydanticOutputParser)OutputParserLike)JsonOutputKeyToolsParserPydanticToolsParser)RunnableRunnablePassthrough) RunnableMap)BaseTool)convert_to_openai_tool)is_basemodel_subclass) BaseModelFieldz)%(asctime)s - %(levelname)s - %(message)sz%Y-%m-%d %H:%M:%S)levelformatdatefmtobjreturnc<t|txr t|SN) isinstancetyper')r-s r/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/chat_models/cloudflare_workersai.py_is_pydantic_classr48s c4 ?%:3%??messagescg}|D]}dt|jtr |jndd}t|trd|d<nt|tr9d|d<|j rb|j Dcgc] }|d|dd}}||d <n:t|t rd |d<n$t|trd |d<|j|d <|j||Scc}w) z;Convert LangChain messages to Cloudflare Workers AI format.)rolecontentuserr9 assistantnameargs)r= arguments tool_callssystemtool tool_call_id) r1r:strrrr@rrrCappend)r6cloudflare_messagesmessagemsgrr@s r3(_convert_messages_to_cloudflare_messagesrI<s(*4W__c*JwPR  g| , CK  +%CK!!&-%7%7!'v.Yv=NO %/L!  /"CK  - CK$$   ""3'9(< !s8C!responsec g}d|jdvrO|jddD]6}|jttt |d|d8|S)z$Get tool calls from ollama response.r@resultr=r?)idr=r>)jsonrErrDr)rJr@tcs r3_get_tool_calls_from_responserPcsnJx}}x00--/(+L9 B   57|FK   r5z0.3.23z1.0z,langchain_cloudflare.ChatCloudflareWorkersAI)sinceremovalalternative_importceZdZUdZedZeed<edZeed<edZ eed<dZ eed<dZ eed<d Z eed <d Z eed <d eddffd Z d deedeeedeed edef dZdeeeeefeedefefd edeeefffd Zddddeeee fde!dee"dd edeeeee fff dZ#e$defdZ%xZ&S)!ChatCloudflareWorkersAIz+Custom chat model for Cloudflare Workers AI. account_id api_tokenmodelr8 ai_gatewayurlz-https://api.cloudflare.com/client/v4/accountsbase_urlz$https://gateway.ai.cloudflare.com/v1 gateway_urlkwargsr.Nc t|di||jr;|jd|jd|jd|j |_y|jd|jd|j |_y)z&Initialize with necessary credentials./z/workers-ai/run/z/ai/run/N)super__init__rYr\rVrXrZr[)selfr] __class__s r3rbz ChatCloudflareWorkersAI.__init__s| "6" ??##$Adoo%6a??##3DJJ<A H --$//):(4::,ODHr5r6stop run_managerc Ft|}dd|ji}djd|D}|d|vr|dndd|jD cic] \}} |dvs || c} }} | dt | dt s | dg| d<t jd | tj|j|| } t| } tt| jtt |  } t#| }t%|g Scc} }w)z3Generate a response based on the messages provided. AuthorizationzBearer  c3tK|]0}d|dd|dd|vrd|dndzd|vrd|dndz2y w) zrole: r9z , content: r:r@z , tools: r8rCz, tool_call_id: Nr`).0rHs r3 z4ChatCloudflareWorkersAI._generate..ss  S[MS^,< =2>#2E3|,-.2 O"S(#3~#6"78    s68toolsN)promptrm)rmz)Sending prompt to Cloudflare Workers AI: )headersrN)r:r@)rG) generations)rIrWjoinitemsr1list_loggerinforequestspostrZrPrrDrNr rrr)rcr6rerfr]formatted_messagesrornkeyvaluedatarJr@ ai_messagechat_generations r3 _generatez!ChatCloudflareWorkersAI._generates&FhO"gdnn-=$>?  *   (/6(9VG_t -3LLNSjc5c>RsEzS  = $ZW t-L!']ODM @GH==7F28<  (T.*5U )<&788Ts  D"Drmc `|Dcgc] }t|}}t|dd|i|Scc}w)z'Bind tools for use in model generation.rmr`)r&rabind)rcrmr]rBformatted_toolsrds r3 bind_toolsz"ChatCloudflareWorkersAI.bind_toolss= EJJD1$7JJw|z@ChatCloudflareWorkersAI.with_structured_output..r5)parsed parsing_errorcyr0r`rs r3rz@ChatCloudflareWorkersAI.with_structured_output..rr5)rr) exception_key)r)pop ValueErrorr4r&rr!r rrrr#assignrwith_fallbacksr$) rcrrrr]ris_pydantic_schema tool_namellm output_parser parser_assign parser_noneparser_with_fallbacks r3with_structured_outputz.ChatCloudflareWorkersAI.with_structured_outputs JJx & >vhGH H/7 ] "(F ' '~ %/v6zB6JI//6( /BC!2E!($(3 !9&! { "))V],C)DC&%V<%'  ++1(!5  /66!%(=8M.44NKK#0#?#? _$@$ 3'*>> >& &r5cy)z9Return the type of the LLM (for Langchain compatibility).zcloudflare-workers-air`)rcs r3 _llm_typez!ChatCloudflareWorkersAI._llm_types'r5)NN)'__name__ __module__ __qualname____doc__r)rVrD__annotations__rWrXrYrZr[r\rrbrrr rrr~r r rr rr%r"rrrr(boolrrpropertyr __classcell__)rds@r3rUrUrs 6CjJ 3ZIssE3JCMCHcC=K= P P P%):> )9{#)9tCy!)967 )9  )9  )9V=d38ndHS#X4FPQR== $i/ 0 ="EW :'dDO+,:' :' !@AB :'  :' $eD)O&<< =:'x'3''r5rU)Hloggingoperatorrtypingrrrrrr r r r r uuidrrvlangchain_classic.schemarrrrlangchain_core._api.deprecationrlangchain_core.callbacksrlangchain_core.language_modelsr*langchain_core.language_models.chat_modelsrlangchain_core.messagesrrrrrlangchain_core.messages.toolrlangchain_core.output_parsersrr"langchain_core.output_parsers.baser*langchain_core.output_parsers.openai_toolsr r!langchain_core.runnablesr"r#langchain_core.runnables.baser$langchain_core.toolsr%%langchain_core.utils.function_callingr&langchain_core.utils.pydanticr'pydanticr(r) basicConfigINFO getLoggerrrtrr4rDrIResponserPrUr`r5r3rs"   XX6==D3@C5)H?% ,, 6  '  H %@C@D@$;$ $sCx.$N H,=,= $x.   E I'mI'  I'r5