MCP AI Server

o ��dh��@s
dd
lZddl
mZ
ddlmZ
ddlmZ
ddlmZm Z
ddl mZ
ddlm Z
ddlmZ
dd lmZ
e�
e d d�
Zed�

e�Zee�
Ze��
Gd d�de�Ze�d�
dd��
Zejddd�defdd��
Zejddd�defdd��
Zej�e�

ej�e�

dddd�ZeZ d
S) �N)
�FastAPI)
� FastApiMCP)
� BaseModel)� EmbedData� Retriever)
�DDGS)
�OpenAI)
�PromptTemplate)
�load_dotenvz gpt-3.5-turbo)
�modelzMCP FastAPI app loaded!c@seZ
dZUeed
<dS)� QueryModel�queryN)�__name__� __module__�__qualname__�str�__annotations__�rr�F:\Project\main.pyrs 
r�
/c�s �
d
di
S)N�status�runningrrrrr�roots�rz$/machine_learning_faq_retrieval_tool�%Search Pinecone vector DB for ML FAQs)
Zsummary�bodyc
�sJ�
td
t
��}
|
�|j�
}td�
}|j||jd�}t�|�
}d|j� �i
S)rZtestmcpzlAnswer the following question based on the context below. Context: {context} Question: {question} Answer:)�contextZquestion�resultN) rr�searchr r �format�llmZcomplete�text�strip)rZ retrieverr�promptZfull_prompt�responserrr�#machine_learning_faq_retrieval_tools�


�

r$z/pinecone_web_search_tool�Search web using DuckDuckGoc
�s|�
g}
t��,}|j
|jd
d�D]}|�dd�}|�dd�}|
�|�d|���

qWd�
d|
i
S1s5w



Y
d|
i
S) r%� )
Zmax_results�title�rz: Nr)rr r �get�append)r�resultsZddgs�
rr'Zsnippetrrr�pinecone_web_search_tool+s�




� ��r-zMy ML FAQ + Web Agentz6Search ML FAQs via Pinecone, web search via DuckDuckGozhttp://127.0.0.1:8083)�name�descriptionZendpoint)!�osZfastapirZfastapi_mcprZpydanticrZragrrZduckduckgo_searchrZllama_index.llms.openairZllama_index.corer �dotenvr r�print�appZmcpZmountrr)rZpostr$r-Ztoolsr*Z__mcp_info__Z__mcp__rrrr�<module>
s:







 








�