BCh?ddlZddlmZmZmZmZddlmZddlZddl Z ddlZddl m Z ddl m Z mZejeZGdde Zy)N)DictAnyOptionalList) current_app) AIModelBase)with_redis_query_cachewith_redis_explain_cacheceZdZdZdZfdZdeeefdefdZ e ddedeeefded e eeefdeeeff d Z e d eeefded edeeeffd ZxZS) DeepSeekModelz8DeepSeek model for SQL generation using direct API callsdeepseekct|tjj d|_tjj dd|_d|_d|_|j stjdy t|_ d|_tjd y#t$r+}tjd t|Yd}~yd}~wwxYw) z7Initialize the DeepSeek model with direct HTTP requestsDEEPSEEK_API_KEYDEEPSEEK_MODELz deepseek-chatz,https://api.deepseek.com/v1/chat/completionsFz[DeepSeek API key not configured. Please set DEEPSEEK_API_KEY in your environment variables.NTzLDeepSeek API integration initialized successfully using direct HTTP requestsz-Error initializing DeepSeek API integration: )super__init__rconfiggetapi_keymodelapi_url initializedloggerwarningrequestsinfo Exceptionstr)selfe __class__s */var/www/html/app/models/deepseek_model.pyrzDeepSeekModel.__init__s "))--.@A  ''++, )r isinstancelistappendjoinitemsextend)rr#formatted_schemar&r4 table_namer)r5 column_namer/ constraints table_infoformatted_columns column_infor0 is_primary is_foreign ref_table ref_columnconstraints_str index_name index_info columns_strr: index_typerelr=r>r?r@rel_types r"_format_schema_for_promptz'DeepSeekModel._format_schema_for_prompt&s-2. fd # ,"YYvy9 ))Ir2 '''*(>? '' 3%LF"(**VY"?K & :y AI"$K$++b Qyk,JK L!''+ , 99-. .'2ooh&C&I&I&K( ( "J  nnY3G " ,3MMO M( ['OOKC )oomTB (__-=uE (__-=uE  &&}5 + b A E Eg NI!,r!B!F!Fx!PJ Z#**-DYKqQ[P\\]+^_"&&z2"&))K"8"%,,r+a {"_L]]^-_`%,,r+a {-KL+ M.  # #gj\$: ;  # #J /  # #$5 6J&:i+@ '' 3.8.C.I.I.K_*J "&))JNN9b,I"JK *{E BI3<'J$++b Bzl${m[\,]^ _  # #B 'Q( (V k )k/.J  # #$4 5"?3 p WW\2 !ggm4 77:.GGK0 77695 ''"ZL+d8*TUV_U``bckbllm(no p  # #B 'yy)**promptdb_type db_configcD|jsdd|jdS|j|}d}|r&d}|jD]\}}|d|d|dz }d |d |d } d |d |d|d|} dd|jd} |j d| dd| dgdd} |j j|j| | } | jdk(r| j}|jdigdjdijddj}|r@|jjds!d||jd |j d!dd"Sd|xsd#|jd |j d!dd$Sd%| jd&| j}t j#|d||jdd$S#t$$rH}t j#d't'|dd(t'||jdd$cYd)}~Sd)}~wwxYw)*z?Generate SQL query using DeepSeek API with direct HTTP requestsF%DeepSeek API not properly initialized)successerrorrr-zDatabase Configuration: z- z: r.z0You are an expert SQL developer specializing in z databases. Your task is to convert natural language requests into correct and efficient SQL queries. Follow these guidelines: 1. Generate only valid as SQL syntax. 2. Focus on writing efficient queries with proper joins and conditions. 3. CAUTION: Return ONLY the SQL query without explanations or markdown formatting. 4. If you cannot generate a valid query, explain why. 5. Use appropriate table aliases for readability. 6. Consider performance implications of your query design. 7. Respect the database schema provided.Database Type: z Database Schema: z z2 Generate a SQL query for the following request: application/jsonBearer z Content-Type Authorizationsystemrolecontentuser皙?rmessages temperatureheadersjsonchoicesrmessagerl)zi cannotz i'm sorryzsorry,Tr8r7)rb sql_queryr http_successzFailed to generate SQL query)rbrcrryDeepSeek API error:  - z$Error generating SQL with DeepSeek: zFailed to generate SQL query: N)r model_typer[rGrrrpostr status_codertrstriplower startswithtextrrcrr)rr]r#r^r_rIdb_config_infokeyvaluesystem_message user_messagersdataresponseresultrx error_messager s r" generate_sqlzDeepSeekModel.generate_sqls @  99+F 8N'oo/ 8 UBse2eWB"77 8NgYW y!) ,+7)4 /&t||n5  ZZ!n=L9  2 }})) *H##s*!#JJy2$7:>>y"MQQR[]_`ffh Y__%6%A%ABf%g$(%.$(OO#4Btzzl!!D(, $)!*!L.L$(OO#4Btzzl!!D(, #7x7K7K6LCPXP]P] _  ]+$*!__$)    LL?AxH I 9#a&B %   s,C G'$G AG H=HHH explain_datarxc|jsddgdStj|d}d|d}d|d |d |d }d d |jd}|jd|dd|dgdd} |j j |j||} | jdk(r| j} | jdigdjdijddj} g} d} | jd}d}d}|D]%jdjvrdvrtfdtd d!Drs jdd"j}|jD]=}|jd#js#t!|jd#} njj#d$rd%}|d%k(sr1djrd&dd'vr|r| j%|}|ss|d(zz }(|r| j%|| s| g} d)| | | |j&d*|jd+d,Sd-| jd.| j(}t*j-|d|gdS#YxYw#t.$r=}t*j-d/t1|dd0t1|gdcYd}~Sd}~wwxYw)1zLAnalyze database explain plan and provide recommendations using DeepSeek APIFra)rbrcrecommendations)indentz?You are an expert database performance analyst specializing in a$ databases. Your task is to analyze the execution plan of a SQL query and provide recommendations for optimization. Follow these guidelines: 1. Focus on identifying performance bottlenecks in the execution plan. 2. Provide specific, actionable recommendations to improve query performance. 3. Consider indexes, join methods, table scans, and other relevant factors. 4. Explain why each recommendation would improve performance. 5. Rate the overall efficiency of the query on a scale of 1-10. 6. Structure your analysis in a clear, organized manner.rdz SQL Query: z Execution Plan: z Please analyze this execution plan and provide: 1. A summary of how the query is being executed 2. Identification of any performance bottlenecks 3. Specific recommendations for optimization 4. An overall efficiency rating (1-10)rerfrgrirjrmrnrorrrurvrrwrlr-r.Nrating:c38K|]}t|vywN)r).0ilines r" z5DeepSeekModel.analyze_explain_plan..@sGm[\ARVGms z.,)recommendationr suggested optimizationrz. r,Tr8r7)rbperformance_analysisrefficiency_ratingrrzr{z,Error analyzing explain plan with DeepSeek: z Failed to analyze explain plan: )rrtdumpsrrrr}rr~rrsplitranyrangeisdigitintrrEr|rrrcrr)rrr^rx explain_textrrrsrrr analysis_textrrlinescurrent_sectioncurrent_recommendation rating_textwordrr rs @r"analyze_explain_planz"DeepSeekModel.analyze_explain_plansr @#% zz,q9 ]]d\ef9<+7)4  ' * /&t||n5  ZZ!n=L9  W }})) *H##s*!!' 9rd ;A > B B9b Q U UV_ac d j j l #%$%!&++D1"&)-&!AD::rs@,, 0R   8 $oKor\