'iYadZddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZ ddlmZeej$j$dz Zee ddlmZd Z dd lmZmZd ZeGd d Z GddZ!eGddZ"GddZ#dZ$e%dk(re$yy#e$rY^wxYw#e$r d ZddlZddlZYjwxYw#e$rd ZdZYnwxYw)aI GENESIS GEMINI EXECUTOR ======================= Direct Gemini API execution for Hyperdrive mode. Uses $300 Google AI Studio credits. No Antigravity proxy needed - direct API calls. Models (January 2026 - Updated): - gemini-2.0-flash: Primary workhorse (2K RPM, 4M TPM, Unlimited RPD) - gemini-2.5-flash: Newer model (1K RPM, 1M TPM, 10K RPD) - gemini-2.5-pro: Complex reasoning (150 RPM, 2M TPM, 10K RPD) - gemini-2.0-flash-lite: High throughput simple tasks (4K RPM) Integrated with GeminiRateMaximizer for intelligent rate limit management. Usage: executor = GeminiExecutor() result = executor.execute("Implement feature X") print(result.response) # With rate limit awareness executor = GeminiExecutor(use_rate_maximizer=True) result = executor.execute_optimized("Analyze this code", task_type="code_review") N)datetime timedelta)Path)DictOptionalAnyList) dataclass) load_dotenvz.envTF)GeminiRateMaximizerTaskTypecleZdZUdZeed<eed<eed<eed<eed<eed<eed<d Z e eed <y ) GeminiResponsezResponse from Gemini execution.successresponsemodel tokens_used cost_estimateexecution_time task_completeNerror) __name__ __module__ __qualname____doc__bool__annotations__strintfloatrr-/mnt/e/genesis-system/core/gemini_executor.pyrr=s:) MM JE8C=r"rceZdZdZgdZddddddZd d d d d d d dZdddddddddd Zd-dedede fdZ de efdZ d.dededede dededefdZ d/dededede dede dedefd Zde efd!Zd0d"e dedefd#Z d1dededede dededefd$Zdededede dedef d%Zd&ed'efd(Z d2d)ed*edefd+Zdefd,Zy)3GeminiExecutorz Direct Gemini API executor for Genesis Hyperdrive. Uses $300 in Google AI Studio credits. No proxy needed - direct API calls. )GEMINI_API_KEYGOOGLE_API_KEYzJE:/genesis-system/Credentials/GoogleAIStudio-Gemini-AgileAdapt-API-KEY.txtgemini-3-flash-previewgemini-2.5-flashgemini-3-pro-previewgemini-2.0-flash-litegemini-2.0-flash-exp)flashflash2proliteexpg?g333333?g@g333333?g333333?)zgemini-2.0-flashr(r*r)zgemini-2.5-pror+r,) researchcode_generation code_review architecturesimple_extractionclassification summarizationagentic_vision deep_visionNapi_key default_modeluse_rate_maximizerc|xs|j|_|jj|||_t d|_d|_d|_|rtr t|_tr.|jr!tj|jyyy#t$r}td|Yd}~Qd}~wwxYw)Nz)E:/genesis-system/data/gemini_usage.jsonlz.Warning: Could not initialize rate maximizer: r;) _load_api_keyr;MODELSgetr<rusage_log_path total_spentrate_maximizerRATE_MAXIMIZER_AVAILABLEr ExceptionprintGENAI_AVAILABLEgenai configure)selfr;r<r=es r#__init__zGeminiExecutor.__init__zs6$"4"4"6 ![[__]MJ"#NO# ": L&9&;# t|| OODLL 1 ,? LFqcJKK Ls"B&& C/CCreturncT|jddD]'}tjj|}|s%|cSt |jd}|j rG|j j}d|vr#|jdddjS|Sy)z&Load API key from environment or file.N=) API_KEY_SOURCESosenvironrCrexists read_textstripsplit)rMenv_varkeykey_filecontents r#rAzGeminiExecutor._load_api_keys++BQ/ G**..)C   ,,Q/0 ?? ((*002Gg~}}S!,Q/5577Nr"promptr system_prompt max_tokens temperaturecached_content_namec |r|jj||n |j}tj}|jst dd|dddddS t r|j||||||} n|j|||||} tj|z } t| jdz} | dz |jj|dz} d | v} t d | |t| | | | }|j||d d |xj| z c_|jrQt|jdz}|rd}|jj!|t|t| d |S#t"$r;}t dd|ddtj|z dt%|cYd }~Sd }~wwxYw)a Execute a prompt with Gemini. Args: prompt: The task/prompt to execute model: Model to use (flash, pro, exp, or full model name) system_prompt: Optional system instruction max_tokens: Max output tokens temperature: Sampling temperature cached_content_name: Optional Titan Memory name to use Returns: GeminiResponse with results FrzNo API key configured)rrrrrrrr?i@B? TASK_COMPLETET)rrrrrrrNd) input_tokens output_tokensr)rBrCr<timer;rrJ_execute_with_sdk_execute_with_restlenr[COSTSr _log_usagerErF record_usagerHr)rMr`rrarbrcrd model_name start_timerrtokens_estimatecostrresultrkrNs r#executezGeminiExecutor.executes.7>*d3SSD,x7M#! 0"-+F OOFF4CL 1    $ """6<<>2S8 &#$L##00!$\!2"%o"6 1M ! #yy{Z7#!f   sD.F G0G G G task_typepriorityc,tt|jdz|z}|jrtrd} t r t |} |jj|| xst j|} | j} | jdkDrFtj| jn&|jj||j} |j!|| ||||S#t $rt j} YwxYw)a Execute with intelligent model selection and rate limit awareness. Uses the GeminiRateMaximizer to select the best available model based on current utilization and task type. Args: prompt: The task/prompt to execute task_type: Type of task for routing (research, code_generation, etc.) system_prompt: Optional system instruction max_tokens: Max output tokens temperature: Sampling temperature priority: Request priority (1-10, lower = higher) cached_content_name: Optional Titan Memory name to use Returns: GeminiResponse with results rgN)token_estimaterzr{r)r`rrarbrcrd)rrpr[rFrGr ValueErrorGENERALschedule_requestr delay_secondsrmsleep TASK_ROUTINGrCr<ry) rMr`rzrarbrcr{rdr}task_type_enum scheduledrs r#execute_optimizedz GeminiExecutor.execute_optimizeds :S0367*D   #;!N6%-i%8N ++<<-(P)%%0k)&&IYbYgYgh%%b) * @ A# /gglll2JJqMqvv.Agglll2 / 2<.E FMM//66)$Z'k2 7EZZ"//kk 1 12!00BB'",    (, - s&C;HA HB!H H! HH!rc,tsy tjjj |}t |j |j|jt|j|jjS#t$rYywxYw)z"Retrieve an existing memory block.Nrrrrr)rJrKrrrCrrrrrrrrrH)rMrrs r# get_memoryzTitanMemoryManager.get_memorys{t MM//33D9EZZ"//kk 1 12!00BB    sA=B BBc `tsgSg} tjjj D]f}|j t |j|j|jt|j|jjh |S#t$rY|SwxYw)zList all active memories.r)rJrKrrlistrrrrrrrrrrH)rMmemoriesrs r# list_memoriesz TitanMemoryManager.list_memoriessr  4499;  !&!3!3++ #E$5$5 6 % 4 4 F F !    sBB B-,B-)Nzgemini-2.0-flash-001r)rrrrrrOr rrrrrrr r!r"r#rr>s22!+ >Dz>> >  > +  >@ s x '< tK0r"rcLt}td|js tdytd|jddd|jddtd|jtd |j d d }td |j td|j ddtd|jtd|jdtd|jddtr+tdt|j}td|jrtd|jtd|j}|jD]\}}t|d|y)zQuick test of Gemini execution.z=== GEMINI EXECUTOR TEST ===\nzERROR: No API key found!Nz API Key: z...zDefault Model: z\n--- Testing Simple Prompt ---z*What is 2 + 2? Reply with just the number.r-)rz Success: z Response: rjzTokens: zCost: $z.4fzTime: z.2fsz)\n--- Testing Titan Memory Capability ---zTitan Manager initialized.zError: z\n--- Budget Status ---r)r%rIr;r<ryrrrrrrJrrritems)executorrxtitanstatusr]values r# test_geminirsH +,    () Ih&&s+,C0@0@0E/F GH OH223 45 ,-   4F  Ifnn% &' Jvt,- ./ HV''( )* GF((- ./ F6((-Q /0 :;"8#3#34 *+||  ~&' $%  ' ' )Flln! U Rw !r"__main__)&rrVrrmrrpathlibrtypingrrrr dataclassesr dotenvr __file__r _env_path ImportErrorgoogle.generativeai generativeairKrJurllib.requestr urllib.errorcore.gemini_rate_maximizerr r rGrr%rrrrr!r"r#r"s4 (,,! "X%%,,v5I  'OH#        b b J     hhV*!Z zM]  O$Hs5-B4B?' C4B<;B<?CC C C