UhBddZddlZddlmZmZmZddlmZddlm Z ddl m Z GddZ dS) zgOpenAI-powered restaurant website generation with OpenRouter provider config and LangSmith integration.N)DictAnyAsyncGenerator) AsyncOpenAI) wrap_openai)settingsc eZdZdZdZdefdZdeeeffdZ d dedede ed ffd Z d dededeeeffd Z d S) AIWorkflowzMOpenAI-powered workflow for handling AI interactions for restaurant websites.c<ttjdtjtjd|_tjr%tjrt|j|_| |_ | |_ dS)Nzhttps://openrouter.ai/api/v1)z HTTP-RefererzX-Title)api_keybase_urldefault_headers) rropenrouter_api_keyopenrouter_referreropenrouter_titleclientlangsmith_api_keylangsmith_tracingr_load_system_prompt system_prompt_build_provider_configprovider_configselfs D/var/www/html/web-builder-api.evdpl.com/src/workflows/ai_workflow.py__init__zAIWorkflow.__init__ s!/3 ( <#4      % 3(*D 3%dk22DK"5577 $::<<returnc tjtjt}tj|dd}t |dd5}|cdddS#1swxYwYdS#t$r}td|Yd}~dSd}~wwxYw) z4Load system prompt from markdown file with fallback.promptszsystem_prompt.mdrzutf-8)encodingNz1Warning: Could not load system prompt from file: a&ONLY USE HTML, CSS AND JAVASCRIPT. If you want to use ICON make sure to import the library first. Try to create the best UI possible by using only HTML, CSS and JAVASCRIPT. Also, try to ellaborate as much as you can, to create something unique. ALWAYS GIVE THE RESPONSE INTO A SINGLE HTML FILE.) ospathdirnameabspath__file__joinopenreadstrip Exceptionprint)r current_dir prompt_filefes rrzAIWorkflow._load_system_prompt!s, @'//"'//(*C*CDDK',,{I?QRRKk3999 (Qvvxx~~'' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( @ @ @ IaII J J J@@@@@@ @s<A4B66&B) B6)B--B60B-1B66 CCCcdtjrtjdngtjtjrtjdngtjdiS)z(Build OpenRouter provider configuration.provider,)orderallow_fallbacksignoredata_collection)rprovider_ordersplitprovider_allow_fallbacksprovider_ignoreprovider_data_collectionrs rrz!AIWorkflow._build_provider_config0sk ?G?V^066s;;;\^#+#DAIAYa(288===_a#+#D   rdefaultprompt thread_idNc Kddl}ddlm}djdd|dg fd}d}d}t t d d }||kr"|d z } ||| d{V} | 23d{V} t | d drK| jrDt | jdddr(t | jdjdd} | r| WVd6dS#|j $r/||krdWVYdS| d |zd{VYt$rJ} ||krdt| WVYd} ~ dS| d |zd{VYd} ~ !d} ~ wwxYwdS)a2 Generate AI response with streaming for restaurant website generation. Args: prompt: User prompt (restaurant website requirements) thread_id: Thread ID for conversation persistence Yields: Streaming response chunks rN)partialsystemrolecontentusercKjjjtjtjtjdjd{VS)NT)modelmessages temperature max_tokensstream extra_body) rchat completionscreater agent_modelrKrLr)rJrsr _stream_callz2AIWorkflow.generate_response.._stream_callUsh)5<<*!$0#./ = rrequest_timeout_secondsx)timeoutchoicesdeltarFz+Error processing request: request timed outzError processing request: ) asyncio functoolsrBrgetattrrwait_forrYrZ TimeoutErrorsleepr,str)rr?r@r[rBrS max_attemptsattemptper_request_timeoutrMchunkrFr1rJs` @rgenerate_responsezAIWorkflow.generate_response<s_ %%%%%%$*< = = / /        %h0I3OO $$ qLG &// H[/\\\\\\\\#)*******%ui66*5=*WUZUbcdUegnptMuMu*")%-*:*@)T"R"R"*")MMMM $*'   l**GGGGGFFmmAK000000000   l**?s1vv??????FFFFFmmAK000000000  /%$s7 'C2C8ACE$1E$ E$E;EE$cPK d|jdd|dg}|jjjt j|t jt j|j d{V}|j dj j ddS#t$r}dt|dcYd}~Sd}~wwxYw) aD Generate AI response synchronously for restaurant website generation. Args: prompt: User prompt (restaurant website requirements) thread_id: Thread ID for conversation persistence Returns: Dictionary containing response and metadata rCrDrG)rIrJrKrLrNNr)responseerror)rrrOrPrQrrRrKrLrrYmessagerFr,ra)rr?r@rJrhr1s rgenerate_response_syncz!AIWorkflow.generate_response_syncs "d.@AAF33H "[-9@@*!$0#./ AH%,Q/7?     Q       sA:A?? B% B B% B%)r>) __name__ __module__ __qualname____doc__rrarrrrrrfrlrrr r sWW===* @S @ @ @ @  S#X     #FFFF T " FFFFV#(((( c3h ((((((rr ) rpr#typingrrropenairlangsmith.wrappersr core.configrr rqrrrvsmm ,,,,,,,,,,****** ccccccccccr