o di ?@sUdZddlmZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZddlmZmZmZddlmZdd lmZdd lZdd lZdd lZdd lmZdd lmZe d ddZe edd Z!e!d urse!j"ddgdddZ#e$e%de%ddZ&e%ddZ'ee'Z(ede'ie$e%dde$e%dde$e%d d!d e)e$e%d"d#e)e$e%d$d#e)e$e%d%d#d&Z*ee*d'd(d)id*Z+ia,d+e-d,<ga.d-e-d.<eZ/dmdnd4d5Z0dod9d:Z1dpdej8dcddddzdfdcZ?dgdhZ@eAdikr~eBdje&ejCdkd/e&dld Sd S){a Recommender MCP Server (HTTP transport via fastmcp_http) -------------------------------------------------------- - Exposes Model Context Protocol tools backed by the in-memory RL recommender. - Served over plain HTTP so LangChain/Ollama clients can talk to it easily. Run locally: python mcp_server.py # binds on $PORT or 7860 Then connect from an MCP-compatible client (e.g., the provided Ollama chatbot). ) annotations)AnyDictListOptional) BaseModelField)FastMCPHttpServer)DataController) FeedbackDBProductCatalogUserDemographics)UnifiedRecommendationEnv) MABSimulationN)datetime)Lockrecommender-mcpz&HTTP MCP server for the RL recommender descriptionZ flask_app/GET)methodscCsdddddfS)Nokrhttp)statusserver transportrrrW/Users/divyeshpatel/Desktop/sahana/Recommender/recommender_rl/recommender/mcp_server.py_root%s r ZPORTMCP_SERVER_PORTZ7860Z RECS_DB_PATHzrecommender_system.dbZdb_pathZ RECS_MAX_ARMSZ50ZRECS_MAX_STEPSZ100ZRECS_N_SUGGESTIONS3ZRECS_USE_CLUSTERING0ZRECS_USE_USER_CONTEXTZRECS_USE_PRICE_OPT)Z data_sourcemax_armsZ max_stepsZ n_suggestionsseedZuse_clusteringZuse_user_contextZuse_price_optimizationepsilon_greedyepsilon皙?)Z environmentalgorithm_typealgorithm_paramszDict[int, int]_PRODUCT_ID_TO_ARM List[int]_ARM_TO_PRODUCT_IDFforceboolreturnNonec CstBtr|s WddSztjtjd}dd|D}Wn ty+g}Ynw|dtjaddttDaWddS1sGwYdS)z> Ensure the product-id<->arm mappings are up to date. NlimitcSs g|] }d|vrt|dqS) product_id)int).0prrr Vs z0_refresh_product_arm_mapping..cSsi|]\}}||qSrr)r6idxpidrrr [sz0_refresh_product_arm_mapping..) _MAPPING_LOCKr+data_controllerget_all_products_ENVr$ Exceptionr- enumerate)r.productsZ product_idsrrr_refresh_product_arm_mappingJs "rCr:r5 Optional[int]cCs|tvr tddt|S)z)Map a product ID to its bandit arm index.Tr.)r+rCget)r:rrr_product_id_to_arm]s  rGarm_idxcCs2tstd|krttkrt|SdSdS)z-Map an arm index back to the real product ID.rN)r-rClen)rHrrr_arm_to_product_idcs rJr3cCstddtd|S)z(Get available product IDs from database.TrEN)rCr-r2rrr_available_products_from_dbks  rKc@seZdZUeddZded<eddZded<eddZded<ed d d Zd ed <edddZ ded<edddZ ded<dS) ProductDatazHuman-readable product namerstr product_namezBrand or maker brand_namez/Top-level category (e.g., apparel, electronics)categoryrzUnit price in USD)gerfloatpriceNzOptional long descriptiondefaultr Optional[str]rzOptional image URL image_url) __name__ __module__ __qualname__rrN__annotations__rOrPrSrrWrrrrrLss rLc@speZdZUeddZded<edddZded <edd dZded <edd dZded <edddZ ded<dS)UserDatazUser's display namerrMnameNzOptional genderrTrVgenderzOptional age group or stringagezOptional emailemailz Optional list of preference tagszOptional[List[str]] preferences) rXrYrZrr]r[r^r_r`rarrrrr\{s r\c@s`eZdZUeddZded<eddddd Zded <ed d d Zded<eddd Zded<dS)RecommendationRequestzUnique user idrr5user_id2zHow many items to recommend)rUrQlern_recommendationsr&zBandit / policy typerTrMr)NzAlgorithm hyperparameterszOptional[Dict[str, Any]]r*) rXrYrZrrcr[rhr)r*rrrrrbs rbc@s<eZdZUded<ded<eddZded<dZd ed <dS) FeedbackDatar5rcitem_idz.One of: view, cart, purchase, remove_from_cartrrM event_typeNrV timestamp)rXrYrZr[rrkrlrrrrris riget_system_statuszget status of the MCP serverDict[str, Any]c sVzddtttdWSty*}zdd|dWYd}~Sd}~ww)z- Return basic server status from DB. rr)rrusersrBeventserrorzstatus failed: rmessageN)r=Zget_user_countZget_product_countZget_total_feedbackr@)errrrms create_userz9Handles the resources can be used for creating a new useruserc sz%td|j|j|jdd}t|}|dkrdddWSd|t|d WSty@}zdd |dWYd}~Sd}~ww) z Create a user (DB). Nr)rcr]r^r_Zuser_cluster_idrqzFailed to create userrrsuccess)rrcrvzcreate_user failed: )r r]r^r_r=Zadd_userZget_userr@)rvrowuidrtrrrrus"   get_usersz6can get list of users has a param limit = 50 (default)rfc sVztj|d}dt||dWSty*}zdd|dWYd}~Sd}~ww)z List users (DB). r2rx)rZ total_usersrorqzget_users failed: rrN)r=Z get_all_usersrIr@)r3rortrrrr{s create_productz9Handles the resources can be used for creating a productsproductc sz*td|j|j|j|j|j|jd}t|}|s dddWSt ddd|j d WSt yE}zdd |dWYd}~Sd}~ww) z Create a product (DB). N)r4rNrOrPrSrrWrqzFailed to create productrrTrErx)rr}zcreate_product failed: ) r rNrOrPrSrrWr=Z add_productrC__dict__r@)r}ryrrtrrrr|s(    get_productsz}can get list of products has a param limit = 50 (default) also has optional params such as category and brand None by defaultrPrVbrandc s~z#|r tj||d}n|rtj||d}ntj|d}dt||dWSty>}zdd|dWYd}~Sd}~ww)z1 List products (DB) with simple filters. r2rx)rZtotal_productsrBrqzget_products failed: rrN)r=Zget_products_by_categoryZget_products_by_brandr>rIr@)rPrr3itemsrtrrrrs submit_feedbackzIHandles the MAB reward by feedback and updates the recommender simulationfeedbackc snzt|j}t|j}t|}|durdd|ddWSt|||jtdd}t |}|s7dddWSt |||j|j durGt j nt |j d g}t|d d d d d}||jd}d} d} tj} | rt| drt| dr|t| jkrt| j|} | dkrt| j|| } d|||| | ddWSty} zdd| dWYd} ~ Sd} ~ ww)z Record a user interaction event (DB + MABSimulation update). This updates the MABSimulation recommender with the feedback. NrqzProduct z! is not available as a bandit armrrre)rcr4rksessionZ active_userz#Failed to save feedback to database)rcrjrkrl?gQ?r(g)ZpurchaseZcartviewZremove_from_cartgr arm_counts arm_rewardsrxz,Feedback processed and MABSimulation updated)rrjrcrewardnqrszsubmit_feedback failed: )r5rjrcrGr rkrutcnowr=Z add_feedbackpdZ DataFramerlZ Timestampnow _SIMULATIONZupdate_algorithmrFcurrent_algorithmhasattrrIrrRrr@) rr:rzZ arm_indexZfbrZevent_dfZreward_mappingrrr algorithmrtrrrrs\      get_recommendationszFGet personalized recommendations using MABSimulation recommender classrequestc slz|jpd}|jpi}|tjks|rt||z tjd|jid\}}Wnty8t\}}Ynwt tj d}|sGdddWSdd |D}d d |D}|s]dd dWSt d t t |jt|}tjtj|j|||d }dd |D} dd | D} | sdddWSg} tj} | rt| drt| drtjdddtj| j| jt| j| jdkd} Wdn1swY| D]'} t| }|dur|t| krt dt dt| |}| |q| dqndgt| } d|jt| tr| n| tjtjrtjjj nd| t| ddWSty5}zdd|dWYd}~Sd}~ww) z Get personalized product recommendations using the MABSimulation recommender. This uses the UnifiedRecommendationEnv and MABSimulation classes for intelligent recommendations. r&rc)optionsr2rqzNo available products in DBrrcSg|]}t|qSr)rGr6r:rrrr8Az'get_recommendations..cSg|]}|dur|qSNrr6Zarmrrrr8Dz)No valid bandit arms mapped from productsre)rrcrhZinput_arms_idxobscSrr)rJrrrrr8UrcSrrrrrrrr8Xrz3Algorithm returned no valid product recommendationsrrignore)divideinvalidr)outwhereNr(rg?gffffff?rxz-Recommendations generated using MABSimulation)rrcrecommendationsr)Zalgorithm_classconfidence_scoresZtotal_recommendationsrszget_recommendations failed: )!r)lowerr*rZ set_algorithmr?resetrcr@rKr$maxminr5rhrIZ invoke_agentrrnpZerrstaterrrZ zeros_likerGrRappend isinstancelisttolist __class__rX)rZ algo_typeZ algo_paramsrinfoZproduct_candidatesZarm_candidatesrhZrecommended_armsrrrZ avg_rewardsr:rHZ confidencertrrrr(s            cCstdurtdtS)zz Expose the underlying Flask app so external servers (uvicorn, gunicorn) can host the FastMCP HTTP transport. Nz,FastMCPHttpServer did not expose a Flask app) _FLASK_APP RuntimeErrorrrrrhttp_appsr__main__u"🚀 Starting MCP HTTP server on :z0.0.0.0)hostZregister_serverport)F)r.r/r0r1)r:r5r0rD)rHr5r0rD)r3r5r0r,)r0rn)rvr\r0rn)rf)r3r5r0rn)r}rLr0rn)NNrf)rPrVrrVr3r5r0rn)rrir0rn)rrbr0rn)D__doc__ __future__rtypingrrrrZpydanticrrZfastmcp_http.serverr Zdbr Z db.modelsr r r Z'rl_recommender.UnifiedRecommendationEnvrZrl_recommender.simulationrZpandasrZnumpyrosr threadingrrgetattrrrouter r5getenvr!ZDB_PATHr=r/r?rr+r[r-r<rCrGrJrKrLr\rbriZtoolrmrur{r|rrrrrXprintZrun_httprrrrs                           @ Y