qi*fDUdZddlZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZmZ ddlmZe ej&j&dz Zeed Zd d d d d d d ZddddddddddZe ee eeffed<d ZdZdZdZdZ dZ!dhZ"hdZ#eGddZ$eGddZ%GddZ&da'ee&ed <d!e&fd"Z(d#ed!e$fd$Z)d%eed!e%fd&Z*d!e+fd'Z,e-d(k(r"ddl.Z.e,Z/e.j`e/rdnd)yy#e$rYwxYw)*a GENESIS KIMI EXECUTOR ====================== Moonshot AI Kimi K2.5 as a Genesis execution backend. Uses Moonshot's OpenAI-compatible API directly (not via OpenRouter). Base URL: https://api.moonshot.cn/v1 Models: moonshot-v1-8k, moonshot-v1-32k, moonshot-v1-128k Same interface as gemini_executor: execute_task_sync() + execute_tasks_parallel(). Model routing: - moonshot-v1-8k (short tasks, rapid Q&A, classification) $0.15/1M tokens - moonshot-v1-32k (typical Genesis stories, code gen) $0.15/1M tokens - moonshot-v1-128k (long documents, multi-file, deep research) $0.60/1M tokens Usage: executor = KimiExecutor() result = executor.execute_task_sync("What is 2 + 2?") print(result.response) # Parallel swarm results = executor.execute_tasks_parallel([ "Analyse revenue pipeline", "Draft outreach email for TradiesVoice", "Research competitor pricing", ]) for r in results: print(r.response) Author: Genesis System Version: 1.0.0 N) dataclassfield)datetime)Path)AnyDictListOptional) load_dotenvz.envzhttps://api.moonshot.cn/v1moonshot-v1-8kmoonshot-v1-32kmoonshot-v1-128k)8k32k128kfaststandardmax333333?inputoutputg333333?)r r rMOONSHOT_PRICING2 g?gN@>rceZdZUdZeed<eed<eed<eed<eed<eed<eed<d Z e eed <d Z eed <d Z eed <defdZ y )ExecutionResultz Unified result from Kimi execution. Mirrors the GeminiResponse pattern so the two executors are interchangeable in the Genesis Execution Layer. successresponsemodel tokens_used cost_estimateexecution_time task_completeNerrorr prompt_tokenscompletion_tokensreturnc|jSN)r%selfs +/mnt/e/genesis-system/core/kimi_executor.py__str__zExecutionResult.__str__qs }})__name__ __module__ __qualname____doc__bool__annotations__strintfloatr+r r,r-r4r5r3r#r#]s_ MM JE8C=M3sr5r#ceZdZUdZeeZeee d<dZ e e d<dZ e e d<dZe e d<dZe e d <dZe e d <ed eefd Zed eefd Zed efdZy)KimiSwarmResultz+Aggregate result from a parallel swarm run.)default_factoryresultsr total_tokens total_costelapsed_seconds success_count failure_countr.cT|jDcgc]}|jc}Scc}w)z9All response texts (including failures as empty strings).)rCr%r2rs r3 responseszKimiSwarmResult.responsess %)LL1q 111s%cn|jDcgc]}|js|jc}Scc}w)zOnly successful response texts.)rCr$r%rKs r3successful_responsesz$KimiSwarmResult.successful_responsess&%)LL>qAII >>>s22c |jdk(S)Nr)rIr1s r3 all_successzKimiSwarmResult.all_successs!!Q&&r5N)r6r7r8r9rlistrCr r#r;rDr=rFr>rGrHrIpropertyr<rMrOr:rQr?r5r3rArAus5%*4%@GT/ "@L#J OU M3M3 24922?d3i??'T''r5rAcjeZdZdZddefdeededefdZe deefd Z de fd Z d Z e d ed ededefdZe dedefdZ dded eedeedededef dZ d deed eedeedededeedefdZdedefdZdededdfdZdeeeffdZdeeeffdZy)! KimiExecutoru` Direct Moonshot API executor for Genesis. Provides the same interface as GeminiExecutor: - execute_task_sync(task) — synchronous single call - execute_tasks_parallel(tasks) — parallel swarm execution Thread-safe; safe to instantiate multiple times (each instance holds its own OpenAI client and usage counters). Nrapi_key default_model max_workerscD|xs|j|_tj|t|_t |t|_td|_ |jjjddd|_ d|_d|_d|_y)Nz0E:/genesis-system/data/kimi_executor_usage.jsonlT)parentsexist_okrrE) _load_api_keyrVMOONSHOT_MODELSget DEFAULT_MODELrWminMAX_CONCURRENT_WORKERSrXrusage_log_pathparentmkdir total_callsrDrF_client)r2rVrWrXs r3__init__zKimiExecutor.__init__s 6$"4"4"6 ,00 N{,BC"#UV ""(((E r5r.czdD]6}tjj|dj}|s4|cSy)z9Load API key from environment (prefers MOONSHOT_API_KEY).)MOONSHOT_API_KEY MOONSHOT_KEYN)osenvironr^strip)varkeys r3r\zKimiExecutor._load_api_keys=8 C**..b)//1C  r5c,t|jS)z(Return True if the API key is available.)r:rVr1s r3 is_configuredzKimiExecutor.is_configuredsDLL!!r5c|j$ ddlm}||j t |_|jS#t$r}t d|d}~wwxYw)z2Lazy-load and return the OpenAI-compatible client.Nr)OpenAIz5openai package not installed. Run: pip install openai)rVbase_url)rfopenairt ImportError RuntimeErrorrVMOONSHOT_BASE_URL)r2rtexcs r3 _get_clientzKimiExecutor._get_clients] <<  ) " *DL|| "K s= A AAr&r,r-cbtj|ddd}|dz |dz|dz |dzzS)z,Return estimated USD cost for a single call.rri@Brr)rr^)r&r,r-pricings r3_estimate_costzKimiExecutor._estimate_costsJ#&&u.MN Y &''*: : 9,0AA B r5taskcNt|j}|dkDry|dkDryy)u Auto-select model based on task length. Heuristic: < 2 000 words → 8k (cheapest) 2 000–8 000 → 32k > 8 000 → 128k (long context) i@rir r )lensplit)r word_counts r3_select_model_for_taskz#KimiExecutor._select_model_for_tasks-&  %  $r5prompt system_prompt max_tokens temperaturec |js!tdd|xs |jdddddS|rtj ||n|j |}g}|r|j d|d|j d |dtj}d } ttD]8} |j} | jjj|||| } tj|z } | j}|r |jnd}|r |j nd}|r |j"nd}|j%|||}| j&dj(j*xsd}d |v}td ||||| ||| }|xj,dz c_|xj"|z c_|xj.|z c_|j1||d d|cStj|z } tdd|dd| d| S#t2$r}t5|} t7t7|dd dd }|t8vxsd| j;vxsd| v}|t<vxs*|xs&d| j;vxsd| j;v}|r | tdz k(rYd }~t?t@d| zztB}|r tE|d}tjF|Yd }~?d }~wwxYw)u) Execute a single prompt with exponential-backoff retry on rate limits. Args: prompt: The user prompt / task description. model: Model key ("fast", "standard", "max") or full name. If None, auto-selected from prompt length. system_prompt: Optional system message. max_tokens: Maximum output tokens. temperature: Sampling temperature (0–1). Returns: ExecutionResult with response and cost metadata. FrkrrEz7MOONSHOT_API_KEY not set. Add to E:\genesis-system\.env)r$r%r&r'r(r)r*r+system)rolecontentuserN)r&messagesrr TASK_COMPLETET) r$r%r&r'r(r)r*r,r-xr% status_codez rate limit429timeout connectiong@)$rrr#rWr]r^rappendtimerange MAX_RETRIESr{chat completionscreateusager,r-rDr~choicesmessagerrerF _log_usage Exceptionr<getattrRATE_LIMIT_CODESlowerRETRYABLE_CODESr`RETRY_BASE_DELAYRETRY_MAX_DELAYrsleep)r2rr&rrrresolved_modelr start_time last_errorattemptclientr%elapsedrr,r-rDcosttextr*resultrzr is_rate_limit is_retryabledelays r3executezKimiExecutor.executes,!!#"1t11!"#Q     u -,,V4  *,  OOX-H IF;<YY[ $( [)@ "G? "))+!;;2299(%) + :))+ 2 7< 3 3! ?DE$;$;!!5:u11 **>=J[\''*22::@b /4 7 ( !( ,"&#*"/"/&7   A% !!\1!4't 5 M@ "D))+ * "  7 " X %gc:t&DmUYZ  #33+#z'7'7'99+ *  ?2:$: J$4$4$66:$z'7'7'99 $w+/'A,W =O sOE 5!!1 "s 1D3H K0BK+(=K++K0taskscHs tSt|xs jtt t j }dt tffd } tj} | jr_ddl } | jjd5} | jtj|} | j!} dddn| j#|} t j |z }t'd D}t'd| D}t'd | D}t | |z }t| ||||| S#1swYuxYw#t$$rtj|} YwxYw) a Execute multiple tasks in parallel using asyncio + thread pool. Up to 50 concurrent workers (configurable). Tasks are returned in the same order they were submitted. Args: tasks: List of prompt strings. model: Model key or full name (None = auto-select per task). system_prompt: Shared system message applied to every task. max_tokens: Max output tokens per task. temperature: Sampling temperature. max_workers: Override concurrent workers (default: self.max_workers). Returns: KimiSwarmResult with ordered results and aggregate stats. r.cKtjtjdtdtf fd }tj Dcgc] }|| c}d{Scc}w7 w)N task_promptr.c K4d{jd|ffd d{cdddd{S7877 #1d{7swYyxYww)Nc0j|S)N)rr&rrrr)prr&r2rrs r3zYKimiExecutor.execute_tasks_parallel.._run_all.._run_one..s$dll#$"'*7'1(3 /;/r5)run_in_executor)rlooprr&r2semrrs r3_run_onezGKimiExecutor.execute_tasks_parallel.._run_all.._run_onesb  !%!5!5!, "         sQAAAA AA  AAAA A AA AA)asyncio Semaphoreget_event_loopr<r#gather) rtrreffective_workersrr&r2rrrs @@r3_run_allz5KimiExecutor.execute_tasks_parallel.._run_allsg##$56C))+D C O  !u)E!(1+)EFF F)EFsAA<A5)A<0A:1 A<rNr)rXc34K|]}|jywr0)r'.0rLs r3 z6KimiExecutor.execute_tasks_parallel..s?Q1==?c34K|]}|jywr0)r(rs r3rz6KimiExecutor.execute_tasks_parallel..s?Q?rc3:K|]}|jsdyw)rN)r$rs r3rz6KimiExecutor.execute_tasks_parallel..sA!qyyAAs)rCrDrFrGrHrI)rAr`rXrarrr r#rr is_runningconcurrent.futuresfuturesThreadPoolExecutorsubmitrunrrun_until_completerxsum)r2rr&rrrrXrrr concurrentpoolfuture results_listrrDrFrHrIrs`````` @r3execute_tasks_parallelz#KimiExecutor.execute_tasks_parallelrsw4"$ $  +4++ " J  YY[  G_ 5 G G( 3))+D )''::q:IJT![[hjAF:@--/LJJ $66xzB ))+ *?,?? ?,?? A|AA L)M9  %!#''   JJ  3";;xz2L 3s+*AE;.6E/$E;/E84E;;#F! F!c *|jdd|i|S)z Synchronous single-task execution. Matches the genesis_execution_layer.execute_task_sync() signature. rr?r)r2rkwargss r3execute_task_synczKimiExecutor.execute_task_syncs t||242622r5rprompt_previewc tjj|j|j|j |j t|jdt|jd|j|j|d } t|jdd5}|jtj |dzdddy#1swYyxYw#t"$rYywxYw) z-Append a JSONL log entry for budget tracking.) timestampr&r,r-rDcost_usdexecution_time_sr$r*rautf-8encoding N)rnow isoformatr&r,r-r'roundr(r)r$r*openrbwritejsondumpsr)r2rrentryfhs r3rzKimiExecutor._log_usages"113\\#11!'!9!9"..f22A6 %f&;&;Q ?~~#11,   d))3A 3RE*T12 3 3 3   s0C(+(CC(C%!C(%C(( C43C4c |j|jt|j|j|j t |jd|jr#|jddd|jdddSddS)z@Return current executor status (mirrors GeminiExecutor pattern).Nrz...zNOT SET) configuredrWrurXrerDtotal_cost_usdapi_key_preview) rrrWryrXrerDrrFrVr1s r3 get_statuszKimiExecutor.get_statuss,,.!//)++++ --#DOOQ7<<<<#$C RS(9':;    r5cd}d}|jjrYt|jdd5}|D]2} tj|}||j ddz }|dz }4 ddd|dkDr||z nd}t|d |t|d ttjd S#tj tf$rYwxYw#1swYjxYw) zAGet cost tracking (parallel to GeminiExecutor.get_budget_status).rErrLrrrrNrr)total_spent_usdreavg_cost_per_callmodels) rbexistsrrloadsr^JSONDecodeErrorKeyErrorrrRr]values)r2spentcallsrlineravgs r3get_budget_statuszKimiExecutor.get_budget_statuss    % % 'd))3A RD $ 4 0:s!;;    %qyeemc$UA !&sA?1134   !00(;   s.C/B:-C:CCCCC")NN ffffff?)NNr r N)r6r7r8r9DEFAULT_CONCURRENT_WORKERSr r<r=rg staticmethodr\r:rrr{r>r~rr#rr rArrrrrrr r?r5r3rUrUs "&'5 # ,8C="t" " c # # RW   S S  * $'+ | | }|  } |  |  |  | F $'+ %)V CyV }V  } V  V  V c]V  V t3c33#$, DcN " 4S> r5rU_default_executorr.c.t tatSr0)rrUr?r5r3_get_default_executorrs (N r5rc 8tj|fi|S)z Drop-in replacement for genesis_execution_layer.execute_task_sync(). Routes through the Moonshot API with auto model selection. )rr)rrs r3rr"s 5 " 4 4T DV DDr5rc 8tj|fi|S)uk Parallel swarm execution — up to 50 concurrent workers. Example: from core.kimi_executor import execute_tasks_parallel results = execute_tasks_parallel([ "Summarise the Q1 revenue pipeline", "Draft outreach email for TradiesVoice", ]) for r in results.results: print(r.response) )rr)rrs r3rr+s  : " 9 9% J6 JJr5c*tdtdtdt}|js"tdtdtdy|j}td|dtd |d td |d ttd |j ddd}|j rtdtd|j jtd|jtd|jdtd|jddtd|jntd|jyttd|jgdddd}|jrtd |jd!t!|j"d"t%|j"d#D].\}}td$|d%|j j0td&|j&td'|j(dtd(|j*ddnktd)|j,d*|j"D]'}|j rtd+|j)|jd,k\r td-nyttdtd.tdy/)0zu Execute a live API call against Moonshot and print results. Returns True on SUCCESS, False on failure. z<============================================================uKIMI EXECUTOR — LIVE TESTz [FAIL] MOONSHOT_API_KEY not set.z% Add to E:\genesis-system\.env:z MOONSHOT_API_KEY=sk-...Fz API Key : rz Model : rWz Workers : rXz1--- Test 1: Single execution (moonshot-v1-8k) ---z*What is 2 + 2? Reply with just the number.r)rr&rz SUCCESSz Response : z Tokens : z Cost : $z.8fz Latency : z.2fsz Model : z FAILED: z8--- Test 2: Parallel swarm (3 tasks, moonshot-v1-8k) ---)zReply with just the word: ALPHAzReply with just the word: BETAzReply with just the word: GAMMAr)rr&rrXu SUCCESS — /z tasks passedrz Task z: z Total tokens : z Total cost : $z Wall time : u PARTIAL FAILURE — z tasks failedz Error: ru8 (Partial success — rate limit transient, acceptable)zKIMI EXECUTOR TEST: SUCCESST)printrUrrrrr$r%rnr'r(r)r&r+rrQrHrrC enumeraterDrFrGrI)executorstatusr swarm_resultirLs r3_run_live_testr?s  (O '( (O~H  ! ! # 01 78 ./  "F Jv/01 23 Jvo./ 01 Jvm,- ./ G =>   ;F ~~  foo335678 f00123 v33C89: f33C8:; fll^,- 6<<.)* G DE22 3 L |99:!C @T@T "<#:#:3"?@A !,">">s!C1EF &|'A'A&B-PQ%% -A99 !''+, -  % % * L M G (O '( (O r5__main__r)1r9rrrlr dataclassesrrrpathlibrtypingrrr r dotenvr __file__rc _env_pathrwryr]rr<r>r;r_rarrrrrrr#rArUrrrrr:rr6sysokexitr?r5r3r*s D (,, "X%%,,v5I 1   ! #'$7"&$7"&$71$sDe,,- "  5+  . '' '8~ ~ L -18L)0|ECEoE K$s) K/ K(PPf z  B CHH"Q!A  s-DDD