pricer-agent/messaging_planning.ipynb
2025-06-10 12:03:09 +05:30

102 lines
2.3 KiB
Plaintext

{
"cells": [
{
"cell_type": "code",
"execution_count": 2,
"id": "267ad308",
"metadata": {},
"outputs": [],
"source": [
"from dotenv import load_dotenv\n",
"from agents.messaging_agent import MessagingAgent"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "be624713",
"metadata": {},
"outputs": [],
"source": [
"agent = MessagingAgent()"
]
},
{
"cell_type": "code",
"execution_count": 4,
"id": "ab42de85",
"metadata": {},
"outputs": [],
"source": [
"agent.push(\"MASSIVE NEWS!!!\")"
]
},
{
"cell_type": "code",
"execution_count": 6,
"id": "854b721c",
"metadata": {},
"outputs": [],
"source": [
"import chromadb\n",
"DB = \"products_vectorstore\"\n",
"client = chromadb.PersistentClient(path=DB)\n",
"collection = client.get_or_create_collection('products')\n",
"from agents.planning_agent import PlanningAgent"
]
},
{
"cell_type": "code",
"execution_count": 7,
"id": "32ffc905",
"metadata": {},
"outputs": [],
"source": [
"planner = PlanningAgent(collection)"
]
},
{
"cell_type": "code",
"execution_count": 8,
"id": "71b1c9b4",
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"Opportunity(deal=Deal(product_description='The Polti Vaporetto Corded Smart Mop Electric Steam Cleaner features 12 multi-purpose attachments, 5-channel swivel mop, up to 50-PSI adjustable steam control, and a 156\" power cord. The model number is PTNA0018.', price=80.0, url='https://www.dealnews.com/products/Polti-Vaporetto-Corded-Smart-Mop-Electric-Steam-Cleaner/489869.html?iref=rss-f1912'), estimate=306.4524282644685, discount=226.4524282644685)"
]
},
"execution_count": 8,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"planner.plan()"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.9.22"
}
},
"nbformat": 4,
"nbformat_minor": 5
}