@h\/dZddlZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlZe je j e j"eZGdde Ze Gd d ZGd d ZddlmZGd deZGddeZGddeZGddZGddZeZeZ ejCeeejCeeejCeey)z Multi-Agent System for FarmGenius 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)levelc8eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z y ) 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_DOCTORCC:\Users\aadil\OneDrive\Desktop\404\backend\app\core\multi_agent.pyr r s83K)'O'O'OJ%N#M'OKr'r ceZdZUdZeed<eed<eed<eed<dZe eefed<de eeffd Z e d e eefddfd Z y) Messagez,Message object for inter-agent communicationsenderreceivercontent message_typeNcontextreturnc|jj|jj|j|j|j xsidS)zConvert message to dictionaryr+r,r-r.r/)r+valuer,r-r.r/selfs r(to_dictzMessage.to_dict'sDkk'' ++|| --||)r   r'datac x|t|dt|d|d|d|jdiS)zCreate message from dictionaryr+r,r-r.r/r2)r get)clsr7s r( from_dictzMessage.from_dict1sGT(^,tJ/0On-HHY+   r') rrrrr __annotations__rstrr/rr6 classmethodr;r&r'r(r*r*sl6  L"GT#s(^" c3h  T#s(^   r'r*cHeZdZdZd deddfdZdedefd Zd e d e e fd Z y)Agentz*Base agent class with common functionalityN agent_typerAgentCoordinatorc.||_||_i|_yN)rArmessage_handlers)r5rArs r(__init__zAgent.__init__?s$&57r'r.handlerc"||j|<y)z.Register a handler for a specific message typeN)rE)r5r.rGs r(register_handlerzAgent.register_handlerDs.5l+r'messager0cK|j|jvr'|j|j|d{Stjd|jj d|jy7>w)z9Process incoming message and optionally return a responseNAgent z! has no handler for message type )r.rEloggerwarningrAr3)r5rJs r(handle_messagezAgent.handle_messageHsq   4#8#8 8D..w/C/CDWMM M NNVDOO$9$9#::[\c\p\p[qr sNs;A>A<?A>rD) rrrrr rFr=rrIr*rrOr&r'r(r@r@<sD4898;M8 6S686G8Ir'r@c*eZdZdZdfd ZdZxZS)MarketExpertAgentz,Specialized agent for market-related queriesct|tj||j d|j t jd|_yNchatzgemini-1.5-pro) superrFr r#rI handle_chatgenaiGenerativeModel gemini_modelr5r __class__s r(rFzMarketExpertAgent.__init__TsB 00+> fd&6&67!112BCr'cK|jjdd}d}|d|} |jj|}t |dr |j n t |}t|j|j|d S#t$r=}tj|jjd|d|}Yd}~dd}~wwxYww) NrJzYou 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-r9rYgenerate_contenthasattrr_r= ExceptionrMerrorrAr3r*r+r5rJ user_message system_promptpromptgemini_response response_textes r(rVzMarketExpertAgent.handle_chatYs**9b9  k  "?(<.9 ]"//@@HO4;OV4TO00Z]^mZnM??^^!(    ] LLDOO112/!E FXYZX[\M ].&C>B '#C C3C C CCrDrrrrrFrV __classcell__r[s@r(rQrQRs6D  r'rQc*eZdZdZdfd ZdZxZS)WeatherAdvisorAgentz-Specialized agent for weather-related queriesct|tj||j d|j t jd|_yrS) rUrFr r$rIrVrWrXrYrZs r(rFzWeatherAdvisorAgent.__init__rsB 22K@ fd&6&67!112BCr'cK|jjdd}d}|d|} |jj|}t |dr |j n t |}t|j|j|d S#t$r=}tj|jjd|d|}Yd}~dd}~wwxYww) NrJr]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_r`z z ()zError routing message: ) r,rrMrgr3debugr+r.rOrf)r5rJrns r(r|zAgentCoordinator.route_messages   4;; . LL9':J:J:P:P9QR S  LL,W^^-A-A,B$wGWGWG]G]F^^`ahauau`vvwx yW%5%56EEgNN NN  LL21#6 7 sCAC: A>$$X."5?s(A.$A.A,A.A.cDKd|_tjdyw)z(Start processing messages from the queueTzAgent coordinator startedNrrMrr4s r(startzAgentCoordinator.starts  /0 cDKd|_tjdyw)zStop the coordinatorFzAgent coordinator stoppedNrr4s r(stopzAgentCoordinator.stops  /0rrD)rrrrrFr@rr*rr|r rr=rrrrrr&r'r(rBrBs> CEC 7 x7H i#UXcghkmphpcq~BCJ~K"1 1r'rBcbeZdZdZdZdedefdZdedeefdZ dede eeffdZ defd Z y ) ModelContextProtocolz Protocol for maintaining context between agent interactions. This allows agents to share a common understanding of the conversation state. ci|_yrDcontextsr4s r(rFzModelContextProtocol.__init__s (* r' context_idr7c"||j|<y)z"Set context data for a specific IDNrr5rr7s r( set_contextz ModelContextProtocol.set_contexts$( j!r'r0c8|jj|S)z"Get context data for a specific ID)rr9r5rs r( get_contextz ModelContextProtocol.get_contexts}}  ,,r'c||jvrUt|j|tr|j|j|ytj d|y|j ||y)z%Update existing context with new dataz Cannot update non-dict context: N)r isinstancedictupdaterMrNrrs r(update_contextz#ModelContextProtocol.update_contexts\  &$-- 3T: j)006!A*NO   Z .r'c<||jvr|j|=yy)zClear context for a specific IDNrrs r( clear_contextz"ModelContextProtocol.clear_context s  & j) 'r'N) rrrrrFr=rrrrrrrr&r'r(rrs] +)c))-c-hsm-//DcN/**r'r)"rrtypingrrrrrrloggingenumr dataclassesr json basicConfigINFO getLoggerrrMr r*r@google.generativeai generativeairWrQrtrxrBrrcontext_protocolrr&r'r(rs  ==! ',,'   8 $         :($  < % <+e+Z4141l**B ') ,[9: .{;< ?;78r'