o zj=0@s:dZddlZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlZe je jde eZGddde Ze Gd d d ZGd d d ZddlmZdd lmZGdddeZGdddeZGdddeZGdddZGdddZ eZ!e Z"e!#ee!e!#ee!e!#ee!dS)z Multi-Agent System for CropPilot This module implements a multi-agent system architecture that coordinates specialized agents for crop disease detection, yield prediction, market analysis, and voice interactions. N)DictListAnyOptionalCallableUnion)Enum) dataclass)levelc@s8eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) AgentTypez)Types of specialized agents in the system coordinatordisease_detectoryield_predictormarket_analyzervoice_assistant translatorchat_assistant market_expertweather_advisor crop_doctorN)__name__ __module__ __qualname____doc__ COORDINATORDISEASE_DETECTORYIELD_PREDICTORMARKET_ANALYZERVOICE_ASSISTANT TRANSLATORCHAT_ASSISTANT MARKET_EXPERTWEATHER_ADVISOR CROP_DOCTORr$r$7/var/www/html/CropPilot/backend/app/core/multi_agent.pyr sr c@szeZdZUdZeed<eed<eed<eed<dZe eefed<de eeffd d Z e d e eefddfd d Z dS)Messagez,Message object for inter-agent communicationsenderreceivercontent message_typeNcontextreturncCs"|jj|jj|j|j|jpidS)zConvert message to dictionaryr'r(r)r*r+)r'valuer(r)r*r+selfr$r$r%to_dict's zMessage.to_dictdatac Cs2|t|dt|d|d|d|didS)zCreate message from dictionaryr'r(r)r*r+r-)r get)clsr2r$r$r% from_dict1s   zMessage.from_dict) rrrrr __annotations__rstrr+rr1 classmethodr5r$r$r$r%r&s  r&c@sLeZdZdZddeddfddZded efd d Zd e d e e fddZ dS)Agentz*Base agent class with common functionalityN agent_typer AgentCoordinatorcCs||_||_i|_dSN)r:r message_handlers)r0r:r r$r$r%__init__?s zAgent.__init__r*handlercC||j|<dS)z.Register a handler for a specific message typeN)r=)r0r*r?r$r$r%register_handlerDzAgent.register_handlermessager,csD|j|jvr|j|j|IdHStd|jjd|jdS)z9Process incoming message and optionally return a responseNAgent z! has no handler for message type )r*r=loggerwarningr:r.)r0rCr$r$r%handle_messageHs  zAgent.handle_messager<) rrrrr r>r7rrAr&rrGr$r$r$r%r9<s r9) get_settingsc*eZdZdZdfdd ZddZZS)MarketExpertAgentz,Specialized agent for market-related queriesNcTttj||d|jt}|jr!|jdkr!tj |jdt |j |_ dSNchatYOUR_GEMINI_API_KEY_NOT_SET)api_key) superr>r r!rA handle_chatrHGEMINI_API_KEYgenai configureGenerativeModel GEMINI_MODEL gemini_modelr0r s __class__r$r%r>V zMarketExpertAgent.__init__c |jdd}d}|d|}z|j|}t|dr |jnt|}Wn#tyH}zt |j j d|d|}WYd}~nd}~wwt |j |j |dd S) NrCzYou are MarketExpert, an AI specialized in Indian agricultural markets. Provide accurate, up-to-date advice on prices, trends, and market strategies. Use real data if available. If the user asks about a specific crop or location, be specific. User: text Gemini error: z:Sorry, MarketExpert AI is temporarily unavailable. Error: chat_responser'r(r)r*r)r3rWgenerate_contenthasattrr`r7 ExceptionrEerrorr:r.r&r'r0rC user_message system_promptpromptgemini_response response_texter$r$r%rQ^& zMarketExpertAgent.handle_chatr<rrrrr>rQ __classcell__r$r$rZr%rJTrJcrI)WeatherAdvisorAgentz-Specialized agent for weather-related queriesNcrKrL) rPr>r r"rArQrHrRrSrTrUrVrWrXrZr$r%r>wr\zWeatherAdvisorAgent.__init__c r]) NrCr^zYou are WeatherAdvisor, an AI expert in Indian agricultural weather. Provide accurate weather forecasts, climate advice, and explain how weather impacts farming. Reference real data if available.r_r`razr r#rArQrHrRrSrTrUrVrWrXrZr$r%r>r\zCropDoctorAgent.__init__c r]) NrCr^zYou are CropDoctor, an AI expert in Indian crop health, soil, and disease management. Give actionable, specific advice for crop issues, pest management, and soil health. Reference real data and best practices for Indian agriculture.r_r`raz8Sorry, CropDoctor AI is temporarily unavailable. Error: rbrcrdrir$r$r%rQrpzCropDoctorAgent.handle_chatr(r)r*r+r,csH|jrt|j||||d}|j|IdHStd|jjddS)z-Send message to another agent via coordinatorr-NrDz$ has no coordinator to send messages)r r&r: route_messagerErhr.)r0r(r)r*r+rCr$r$r% send_messageszCropDoctorAgent.send_messager<)rrrrr>rQr rr7rrr&rwrrr$r$rZr%rus 4ruc @sxeZdZdZddZdefddZdedeefd d Z dd e d e de de e e fdeef ddZddZddZd S)r;z4Central coordinator for managing agent communicationcCsi|_t|_d|_dS)NF)agentsasyncioQueue message_queuerunningr/r$r$r%r>s  zAgentCoordinator.__init__agentcCs*||_||j|j<td|jjdS)z&Register an agent with the coordinatorzRegistered agent: N)r rxr:rEinfor.)r0r}r$r$r%register_agents zAgentCoordinator.register_agentrCr,c s|j|jvrtd|jjdSz!td|jjd|jjd|jd|j|j|IdHWSt yO}ztd|WYd}~dSd}~ww)z(Route a message to the appropriate agentzNo agent registered for type: NzRouting message: z -> z ()zError routing message: ) r(rxrErhr.debugr'r*rGrg)r0rCror$r$r%rvs (zAgentCoordinator.route_messageNr'r)r*r+c sVg}|jD] \}}||kr(t|||||d}||IdH} | r(|| q|S)z.Send a message to all agents except the senderr-N)rxitemsr&rGappend) r0r'r)r*r+ responsesr:r}rCresponser$r$r%broadcast_messages  z"AgentCoordinator.broadcast_messagecd|_tddS)z(Start processing messages from the queueTzAgent coordinator startedNr|rEr~r/r$r$r%startzAgentCoordinator.startcr)zStop the coordinatorFzAgent coordinator stoppedNrr/r$r$r%stoprzAgentCoordinator.stopr<)rrrrr>r9rr&rrvr rr7rrrrrr$r$r$r%r;s,  r;c@sheZdZdZddZdedefddZdedeefd d Z dede eeffd d Z defd dZ dS)ModelContextProtocolz Protocol for maintaining context between agent interactions. This allows agents to share a common understanding of the conversation state. cCs i|_dSr<contextsr/r$r$r%r>s zModelContextProtocol.__init__ context_idr2cCr@)z"Set context data for a specific IDNrr0rr2r$r$r% set_contextrBz ModelContextProtocol.set_contextr,cCs |j|S)z"Get context data for a specific ID)rr3r0rr$r$r% get_context s z ModelContextProtocol.get_contextcCsR||jvr!t|j|tr|j||dStd|dS|||dS)z%Update existing context with new dataz Cannot update non-dict context: N)r isinstancedictupdaterErFrrr$r$r%update_context s z#ModelContextProtocol.update_contextcCs||jvr |j|=dSdS)zClear context for a specific IDNrrr$r$r% clear_contexts  z"ModelContextProtocol.clear_contextN) rrrrr>r7rrrrrrrr$r$r$r%rs r)$rrytypingrrrrrrloggingenumr dataclassesr json basicConfigINFO getLoggerrrEr r&r9google.generativeai generativeairSapp.core.configrHrJrtrur;rr context_protocolrr$r$r$r%s4       !!06!