95 lines
21 KiB
Plaintext
95 lines
21 KiB
Plaintext
|
§
|
|||
|
|
/H�h E ã ó$ — d dl Z d dlmZ d dlmZ d dlZd dlmZmZm Z m
|
|||
|
|
Z
|
|||
|
|
d dl m Z d dl
|
|||
|
|
mZ d dlmZmZmZ d dlmZ d d lmZmZmZmZ d d
|
|||
|
|
lmZ d dl Z d dl Z d d l m Z d d l m!Z! d d
|
|||
|
|
l"m#Z# d dl$m%Z% d dl&m'Z' e!¦ « Z(e G d„ d¦ « ¦ « Z) G d„ d¦ « Z* G d„ de ¦ « Z+ G d„ d¦ « Z,e-dk rNdddœd d dœgZ.d „ e.D ¦ « Z/ e,e/d ¬ ¦ « Z0d!Z1e0 2 e1¦ « Z3 e4d"e3d# ¦ « dS dS )$é N)Ú
|
|||
|
|
ChatOpenAI)Ú
|
|||
|
|
ChatAnthropic)ÚListÚDictÚAnyÚOptional)Ú TypedDict)ÚChatPromptTemplate)Ú
|
|||
|
|
StateGraphÚSTARTÚEND)Ú MemorySaver)Ú HumanMessageÚ AIMessageÚ ToolMessageÚ
|
|||
|
|
SystemMessage)Ú dataclass)Úwraps)Úsetup_prompt_manager)Ú LlmConfig)Ú AgenTools)ÚConfigc ó$ — e Zd ZU eed< eed< dS )ÚMessageÚroleÚcontentN)Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__ÚstrÚ__annotations__© ó ú8/Users/macbook/Desktop/blessing_ai/mkd/drone_bot/test.pyr r s" € € € € € € à
|
|||
|
|
|
|||
|
|
€I€I�IØ
|
|||
|
|
€L€L�L€L€Lr# r c óD — e Zd ZdZd defd„Zd„ Zdefd„Zde fd „Z
|
|||
|
|
d
|
|||
|
|
„ Z d S )
|
|||
|
|
ÚModelRotationManagerz+Manages model rotation and failure trackingTÚuse_openai_as_fallbackc óŒ — g | _ |r¨t j j j dt j t j j d¦ « dœt j j j dt j t j j d¦ « dœg}| j |¦ « d„ | j D ¦ « | _
|
|||
|
|
d| _ d| _ || _
|
|||
|
|
t dt | j ¦ « › d�¦ « t d |rd
|
|||
|
|
nd › �¦ « | j D ]$}t d |d
|
|||
|
|
› d|d › d�¦ « Œ%d S )NÚopenaiÚ drone_bot)ÚnameÚproviderÚapi_keyÚ temperaturec ó — i | ] }|d d“Œ S ©r+ r r" ©Ú.0Úmodels r$ ú
|
|||
|
|
<dictcomp>z1ModelRotationManager.__init__.<locals>.<dictcomp>? ó € Ð IÐ IÐ I°E˜u Vœ}¨aÐ IÐ IÐ Ir# r é z&ModelRotationManager initialized with z modelsúOpenAI fallback: ÚEnabledÚDisabledz - r+ ú (r, ú))Úmodelsr r) Úgpt_4or ÚOPENAI_API_KEYÚ temperaturesÚgetÚgpt_4_1ÚextendÚfailure_countsÚcurrent_model_indexÚmax_failures_per_modelr' ÚprintÚlen)Úselfr' Ú
|
|||
|
|
openai_modelsr3 s r$ Ú__init__z ModelRotationManager.__init__ si € ð
|
|||
|
|
ˆŒ ð "ð .õ &Ô ,Ô 3Ô :Ø (Ý %Ô 4Ý#,Ô#3Ô#@×#DÒ#DÀ[Ñ#QÔ#Qð ð õ &Ô ,Ô 3Ô ;Ø (Ý %Ô 4Ý#,Ô#3Ô#@×#DÒ#DÀ[Ñ#QÔ#Qð ð ð
|
|||
|
|
ˆMð
|
|||
|
|
ŒK× Ò ˜}Ñ -Ô -Ð -ð JÐ I¸T¼[Ð IÑ IÔ IˆÔØ#$ˆÔ Ø&'ˆÔ#Ø&<ˆÔ#å
|
|||
|
|
ÐPµs¸4¼;Ñ7GÔ7GÐPÐPÐPÑQÔQÐQÝ
|
|||
|
|
ÐWÐ/EÐ"U ) )È:ÐWÐWÑXÔXÐXØ”[ð @ð @ˆEÝ Ð>˜˜vœ Ð>Ð>¨%°
|
|||
|
|
Ô*;Ð>Ð>Ð>Ñ ?Ô ?Ð ?Ð ?ð @ð @r# c ó& — | j | j S )z#Get the current model configuration)r< rD )rH s r$ Úget_current_modelz&ModelRotationManager.get_current_modelI s € àŒ{˜4Ô3Ô4Ð4r# Ú model_configc óò — |d dk r$t |d |d |d ¬¦ « S |d dk r$t |d |d |d ¬¦ « S t d|d › �¦ « ‚) z3Create an LLM instance based on model configurationr, Ú anthropicr- r+ r. )r- r3 r. r) zUnsupported provider: )r r Ú
|
|||
|
|
ValueError)rH rM s r$ Úcreate_llm_instancez(ModelRotationManager.create_llm_instanceM s € à ˜
|
|||
|
|
Ô # {Ò 2Ð 2Ý Ø$ YÔ/Ø" 6Ô*Ø (¨ Ô 7ðñ ô ð
|
|||
|
|
ð
|
|||
|
|
˜*Ô
|
|||
|
|
%¨Ò
|
|||
|
|
1Ð
|
|||
|
|
1Ý Ø$ YÔ/Ø" 6Ô*Ø (¨ Ô 7ðñ ô ð
|
|||
|
|
õ Ð P°lÀ:Ô6NÐ PÐ PÑQÔQÐ Qr# Úfailed_model_namec ó$ — | j |xx dz
|
|||
|
|
cc< t d|› d| j | › �¦ « d}|t | j ¦ « k r‰| j dz t | j ¦ « z | _ | j | j }| j |d | j k rt d|d › �¦ « |S |dz
|
|||
|
|
}|t | j ¦ « k °‰t d¦ « d„ | j D ¦ « | _ d| _ | j d S ) z Rotate to next model on failureé zModel z failed. Failure count: r r+ zRotating to model: zBAll models have failed. Resetting failure counts and trying again.c ó — i | ] }|d d“Œ S r0 r" r1 s r$ r4 z:ModelRotationManager.rotate_on_failure.<locals>.<dictcomp>r r5 r# )rC rF rG r< rD rE )rH rR ÚattemptsÚ
|
|||
|
|
next_models r$ Úrotate_on_failurez&ModelRotationManager.rotate_on_failure^ s/ € à ÔÐ -Ð.Ð.Ô.°!Ñ3Ð.Ð.Ñ.å
|
|||
|
|
ÐjÐ(ÐjÐjÀ$ÔBUÐVgÔBhÐjÐjÑkÔkÐkð ˆØ�˜Tœ[Ñ)Ô)Ò)Ð)Ø(,Ô(@À1Ñ(DÍÈDÌKÑHXÔHXÑ'XˆDÔ $Ø œ TÔ%=Ô>ˆJàÔ" :¨fÔ#5Ô6¸Ô9TÒTÐTÝÐ@¨J°vÔ,>Ð@Ð@ÑAÔAÐAØ!Ð!à ˜‰MˆHð �˜Tœ[Ñ)Ô)Ò)Ð)õ ÐRÑSÔSÐSØ IÐ I¸T¼[Ð IÑ IÔ IˆÔØ#$ˆÔ ØŒ{˜1Œ~Ð r# c óœ — | ¦ « }| j |d | j k r|S | |d ¦ « S )zGet next available modelr+ )rL rC rE rX )rH Ú
|
|||
|
|
current_models r$ Úget_next_modelz#ModelRotationManager.get_next_modelv sM € à ×.Ò.Ñ0Ô0ˆ
|
|||
|
|
Ø Ô ˜}¨VÔ 4Ô 5¸Ô8SÒ SÐ SØ Ð à×)Ò)¨-¸Ô*?Ñ@Ô@Ð @r# N)T) r r r Ú__doc__ÚboolrJ rL ÚdictrQ r rX r[ r" r# r$ r& r& sŸ € € € € € Ø5Ð5ð'@ð '@¨tð '@ð '@ð '@ð '@ðR5ð 5ð 5ðR°ð Rð Rð Rð Rð" °3ð ð ð ð ð0Að Að Að Að Ar# r& c ón — e Zd ZU eeez ed< ee ed< ee ed< ee ed< dS )ÚStateÚmessagesÚfinal_responseÚ
|
|||
|
|
user_questionrZ N) r r r r r r r! r r r" r# r$ r` r` sX € € € € € € Ø�< )Ñ+Ô,Ð,Ð,Ñ,Ø ˜S”MÐ!Ð!Ñ!ؘC”=Ð Ð Ñ Ø˜C”=Ð Ð Ñ Ð Ð r# r` c ó¨ — e Zd Zddeee defd„Zd„ Zde de
|
|||
|
|
fd „Z de fd
|
|||
|
|
„Z
|
|||
|
|
d ee deee z fd „Zd
|
|||
|
|
e
|
|||
|
|
dee
|
|||
|
|
ef fd„ZdS )ÚDroneBotNTÚhistoryr' c óì — |�|ng | _ || _ t |¬¦ « | _ d| _ d| _ | ¦ « | _ t d¦ « t d|rdnd› �¦ « dS )zÞ
|
|||
|
|
Initialize DroneBot - a simplified plotting agent
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
history: Optional conversation history
|
|||
|
|
use_openai_as_fallback: Whether to include OpenAI models as fallback
|
|||
|
|
N)r' Ú zDroneBot initializedr7 r8 r9 ) rf r' r& Ú
|
|||
|
|
model_managerÚ
|
|||
|
|
final_messageÚfinal_model_usedÚ
|
|||
|
|
_create_toolsÚtoolsrF )rH rf r' s r$ rJ zDroneBot.__init__ˆ s� € ð #*Ð"5�w�w¸2ˆŒ Ø&<ˆÔ#õ 2ÐI_Ð `Ñ `Ô `ˆÔð ˆÔØ "ˆÔ ð ×'Ò'Ñ)Ô)ˆŒ
|
|||
|
|
å
|
|||
|
|
Ð%Ñ&Ô&Ð&Ý
|
|||
|
|
ÐWÐ/EÐ"U ) )È:ÐWÐWÑXÔXÐXÐXÐXr# c ó — t j }|S )zdefine tools for agent)r rm )rH rm s r$ rl zDroneBot._create_tools s € õ ”ˆØˆ r# ÚresponseÚreturnc óŠ — t |d¦ « rLt |j t ¦ « r2|j ¦ « r|j ¦ « S t |d¦ « r…t |j t
|
|||
|
|
¦ « rk|j D ]a}t |t ¦ « rJ| d¦ « dk r1| d¦ « r |d ¦ « c S ŒbdS t |d¦ « r |j rdS dS # t $ r*}t dt |¦ « › �¦ « Y d}~d S d}~ww xY w)
|
|||
|
|
z<Extract the final message content from an AIMessage responser ÚtypeÚtextÚ
|
|||
|
|
tool_callsz Generating your visualization...zProcessing your request...z"Error extracting message content: Nz/I encountered an issue processing your request.) ÚhasattrÚ
|
|||
|
|
isinstancer r ÚstripÚlistr^ r@ rt Ú ExceptionrF )rH ro ÚitemÚes r$ Ú _extract_final_message_contentz'DroneBot._extract_final_message_content§ sv € ð Eå�x Ñ+Ô+ð
|
|||
|
|
4µ
|
|||
|
|
¸8Ô;KÍSÑ0QÔ0Qð
|
|||
|
|
4ÐV^ÔVf×VlÒVlÑVnÔVnð
|
|||
|
|
4Ø Ô'×-Ò-Ñ/Ô/Ð/õ ˜ 9Ñ-Ô-ð
|
|||
|
|
4µ*¸XÔ=MÍtÑ2TÔ2Tð
|
|||
|
|
4Ø $Ô ,ð 4ð 4�DÝ! $Ñ-Ô-ð 4°$·(²(¸6Ñ2BÔ2BÀfÒ2LÐ2LÐQU×QYÒQYÐZ`ÑQaÔQaÐ2LØ # Fœ|× 1Ò 1Ñ 3Ô 3Ð3Ð3Ð3øð4ð 4õ
|
|||
|
|
˜ <Ñ0Ô0ð
|
|||
|
|
4°XÔ5Hð
|
|||
|
|
4Ø9Ð9ð 4Ð3øåð Eð Eð EÝ Ð?µs¸1±v´vÐ?Ð?Ñ @Ô @Ð @ØDÐDÐDÐDÐDÐDøøøøð Eøøøs+ ‚AD Á BD Ã0D Ã3D Ä
|
|||
|
|
EÄ D=Ä=Ec ó& ‡ — t d¦ « t t ¦ « }dt fˆ fd„ }dt fd„} G d„ d¦ « } |‰ j ‰ ¬¦ « }| d|¦ « | d |¦ « | t d¦ « | d|d d t t i¦ « | d d¦ « t ¦ « }|
|
|||
|
|
|¬
|
|||
|
|
¦ « }t d ¦ « |S ) z Create the DroneBot workflowzCreating DroneBot workflowÚstatec ó„ •— ‰
|
|||
|
|
j ¦ « }d}d}t |¦ « D �]} ‰
|
|||
|
|
j |¦ « }t d|d › d|d › d|dz › �¦ « | ‰
|
|||
|
|
j ¦ « }| d }|rt |d
|
|||
|
|
t ¦ « s.t
|
|||
|
|
d ¦ « }t |¬ ¦ « g|z }t d
|
|||
|
|
t |¦ « › �¦ « | |¦ « } t dt | ¦ « j › �¦ « t | d¦ « r-t d| j rt | j ¦ « nd
|
|||
|
|
› �¦ « ‰
|
|||
|
|
| ¦ « }
|
|||
|
|
|
|
|||
|
|
‰
|
|||
|
|
_ d | d | gz i} |d | d< |d ‰
|
|||
|
|
_ | c S # t( $ rx} | }t d|dz › d|d › dt+ | ¦ « › �¦ « ||dz
|
|||
|
|
k r4‰
|
|||
|
|
j |d ¦ « }t/ j d¦ « Y d} ~ �Œ
|
|||
|
|
d} ~ ww xY w|‚)z+Main chatbot that handles plotting requestsé Nz
|
|||
|
|
Using model: r+ r: r, z ) - Attempt rT ra r Úbooking©r zDroneBot input messages: zDroneBot response: rt z Tool calls: rZ zAttempt z failed with model ú: )ri rL ÚrangerQ rF Ú
|