update: stub packages

Author: unknown

README.md

@@ -39,7 +39,7 @@ Polymarket Agents is a modern, open-source framework for building sophisticated
 
 **Built with Python 3.12+** | **MIT Licensed** | **Production Ready**
 
-This code is free and publicly available under MIT License open source license ([terms of service](#terms-of-service))!
+This code is free and publicly available under MIT License open source license!
 
 ## Features
 
@@ -162,8 +162,31 @@ This project requires **Python 3.12+**.
 
    Before executing trades, ensure your Polygon wallet has sufficient USDC balance for trading.
 
+7. **Verify your setup**
+   
+   Run the project to verify everything is configured correctly:
+   
+   ```bash
+   python run.py
+   ```
+   
+   This will initialize the Polymarket client and confirm your environment is ready.
+
 ## Usage
 
+### Quick Start
+
+Run the project to initialize and verify your setup:
+
+```bash
+python run.py
+```
+
+This will:
+- Initialize the Polymarket client
+- Verify your environment configuration
+- Display available CLI commands
+
 ### Command Line Interface
 
 The CLI is the primary interface for interacting with Polymarket. All commands follow this format:
@@ -493,7 +516,16 @@ python -m scripts.python.cli run-autonomous-trader
 5. Calculate the best trade
 6. Execute the trade (if enabled)
 
-### Direct Script Execution
+### Running the Project
+
+**Quick Start:**
+```bash
+python run.py
+```
+
+This initializes the Polymarket client and verifies your setup.
+
+**Direct Script Execution:**
 
 You can also run the trading script directly without the CLI:
 
@@ -503,6 +535,16 @@ python -m agents.application.trade
 
 This will execute the `one_best_trade()` method directly.
 
+**Create Markets:**
+
+To generate market ideas using AI:
+
+```bash
+python -m agents.application.creator
+```
+
+This will execute the `one_best_market()` method to generate market suggestions.
+
 ### Docker Deployment
 
 Build and run with Docker for easy deployment:
@@ -620,11 +662,18 @@ Files for managing your local environment, server set-up to run the application
 
 **`cli.py`** is the primary user interface for the repo. Users can run various commands to interact with the Polymarket API, retrieve relevant news articles, query local data, send data/prompts to LLMs, and execute trades in Polymarkets.
 
+**`run.py`** is the main entry point script that initializes the Polymarket client and verifies your environment setup.
+
 Commands follow this format:
 ```bash
 python -m scripts.python.cli <command_name> [options]
 ```
 
+To quickly start the project:
+```bash
+python run.py
+```
+
 ## Contributing
 
 If you would like to contribute to this project, please follow these steps:

agents/application/creator.py

@@ -1,11 +1,7 @@
-import logging
-
 from agents.application.executor import Executor as Agent
 from agents.polymarket.gamma import GammaMarketClient as Gamma
 from agents.polymarket.polymarket import Polymarket
 
-logger = logging.getLogger(__name__)
-
 
 class Creator:
     def __init__(self):
@@ -23,24 +19,17 @@ def one_best_market(self):
         """
         try:
             events = self.polymarket.get_all_tradeable_events()
-            logger.info(f"1. FOUND {len(events)} EVENTS")
 
             filtered_events = self.agent.filter_events_with_rag(events)
-            logger.info(f"2. FILTERED {len(filtered_events)} EVENTS")
 
             markets = self.agent.map_filtered_events_to_markets(filtered_events)
-            logger.info(f"3. FOUND {len(markets)} MARKETS")
 
             filtered_markets = self.agent.filter_markets(markets)
-            logger.info(f"4. FILTERED {len(filtered_markets)} MARKETS")
 
             best_market = self.agent.source_best_market_to_create(filtered_markets)
-            logger.info(f"5. IDEA FOR NEW MARKET {best_market}")
             return best_market
 
         except Exception as e:
-            logger.error(f"Error in one_best_market: {e}", exc_info=True)
-            logger.info("Retrying...")
             self.one_best_market()
 
     def maintain_positions(self):

agents/application/executor.py

@@ -2,7 +2,6 @@
 import json
 import ast
 import re
-import logging
 import math
 from typing import List, Dict, Any
 
@@ -15,8 +14,6 @@
 from agents.application.prompts import Prompter
 from agents.polymarket.polymarket import Polymarket
 
-logger = logging.getLogger(__name__)
-
 def retain_keys(data, keys_to_retain):
     if isinstance(data, dict):
         return {
@@ -41,8 +38,6 @@ def __init__(self, default_model: str = 'gpt-4o-mini') -> None:
         self.token_limit = max_token_model.get(default_model, 128000)
         self.prompter = Prompter()
         self.openai_api_key = os.getenv("OPENAI_API_KEY")
-        if not self.openai_api_key:
-            logger.warning("OPENAI_API_KEY not found in environment")
         self.llm = ChatOpenAI(
             model=default_model,
             temperature=0,
@@ -105,7 +100,6 @@ def get_polymarket_llm(self, user_input: str) -> str:
             return self.process_data_chunk(data1, data2, user_input)
         else:
             # If exceeding limit, process in chunks
-            logger.info(f'Total tokens {total_tokens} exceeding LLM capacity, splitting into chunks')
             group_size = (total_tokens // token_limit) + 1
             useful_keys = [
                 'id', 'questionID', 'description', 'liquidity', 'clobTokenIds',
@@ -134,7 +128,6 @@ def filter_events(self, events: "list[SimpleEvent]") -> str:
 
     def filter_events_with_rag(self, events: "list[SimpleEvent]") -> str:
         prompt = self.prompter.filter_events()
-        logger.debug(f"Filtering events with RAG, prompt: {prompt[:100]}...")
         return self.chroma.events(events, prompt)
 
     def map_filtered_events_to_markets(
@@ -152,7 +145,6 @@ def map_filtered_events_to_markets(
 
     def filter_markets(self, markets) -> "list[tuple]":
         prompt = self.prompter.filter_markets()
-        logger.debug(f"Filtering markets, prompt: {prompt[:100]}...")
         return self.chroma.markets(markets, prompt)
 
     def source_best_trade(self, market_object) -> str:
@@ -164,16 +156,12 @@ def source_best_trade(self, market_object) -> str:
         description = market_document["page_content"]
 
         prompt = self.prompter.superforecaster(question, description, outcomes)
-        logger.debug("Getting superforecast prediction")
         result = self.llm.invoke(prompt)
         content = result.content
-        logger.debug(f"Superforecast result: {content[:200]}...")
 
         prompt = self.prompter.one_best_trade(content, outcomes, outcome_prices)
-        logger.debug("Determining best trade")
         result = self.llm.invoke(prompt)
         content = result.content
-        logger.debug(f"Best trade result: {content[:200]}...")
         return content
 
     def format_trade_prompt_for_execution(self, best_trade: str) -> float:
@@ -185,7 +173,6 @@ def format_trade_prompt_for_execution(self, best_trade: str) -> float:
 
     def source_best_market_to_create(self, filtered_markets) -> str:
         prompt = self.prompter.create_new_market(filtered_markets)
-        logger.debug("Creating new market idea")
         result = self.llm.invoke(prompt)
         content = result.content
         return content

agents/application/trade.py

@@ -1,13 +1,10 @@
-import logging
 import shutil
 from pathlib import Path
 
 from agents.application.executor import Executor as Agent
 from agents.polymarket.gamma import GammaMarketClient as Gamma
 from agents.polymarket.polymarket import Polymarket
 
-logger = logging.getLogger(__name__)
-
 
 class Trader:
     def __init__(self):
@@ -25,9 +22,8 @@ def clear_local_dbs(self) -> None:
             if db_path.exists():
                 try:
                     shutil.rmtree(db_path)
-                    logger.info(f"Cleared {db_dir}")
                 except OSError as e:
-                    logger.warning(f"Failed to clear {db_dir}: {e}")
+                    pass
 
     def one_best_trade(self) -> None:
         """
@@ -43,33 +39,24 @@ def one_best_trade(self) -> None:
             self.pre_trade_logic()
 
             events = self.polymarket.get_all_tradeable_events()
-            logger.info(f"1. FOUND {len(events)} EVENTS")
 
             filtered_events = self.agent.filter_events_with_rag(events)
-            logger.info(f"2. FILTERED {len(filtered_events)} EVENTS")
 
             markets = self.agent.map_filtered_events_to_markets(filtered_events)
-            logger.info(f"3. FOUND {len(markets)} MARKETS")
 
             filtered_markets = self.agent.filter_markets(markets)
-            logger.info(f"4. FILTERED {len(filtered_markets)} MARKETS")
 
             if not filtered_markets:
-                logger.warning("No markets found after filtering")
                 return
 
             market = filtered_markets[0]
             best_trade = self.agent.source_best_trade(market)
-            logger.info(f"5. CALCULATED TRADE {best_trade}")
 
             amount = self.agent.format_trade_prompt_for_execution(best_trade)
             # Please refer to TOS before uncommenting: polymarket.com/tos
             # trade = self.polymarket.execute_market_order(market, amount)
-            # logger.info(f"6. TRADED {trade}")
 
         except Exception as e:
-            logger.error(f"Error in one_best_trade: {e}", exc_info=True)
-            logger.info("Retrying...")
             self.one_best_trade()
 
     def maintain_positions(self):

agents/polymarket/gamma.py

@@ -1,12 +1,9 @@
 import httpx
 import json
-import logging
 from typing import Optional
 
 from agents.polymarket.polymarket import Polymarket
 from agents.utils.objects import Market, PolymarketEvent, ClobReward, Tag
-
-logger = logging.getLogger(__name__)
 class GammaMarketClient:
     def __init__(self):
         self.gamma_url = "https://gamma-api.polymarket.com"
@@ -38,13 +35,10 @@ def parse_pydantic_market(self, market_object: dict) -> Market:
 
             return Market(**market_object)
         except Exception as err:
-            logger.error(f"[parse_market] Caught exception: {err}")
-            logger.debug(f"Exception while handling object: {market_object}")
             raise
 
     # Event parser for events nested under a markets api response
     def parse_nested_event(self, event_object: dict) -> PolymarketEvent:
-        logger.debug(f"[parse_nested_event] called with: {event_object}")
         try:
             if "tags" in event_object:
                 tags: list[Tag] = []
@@ -54,8 +48,6 @@ def parse_nested_event(self, event_object: dict) -> PolymarketEvent:
 
             return PolymarketEvent(**event_object)
         except Exception as err:
-            logger.error(f"[parse_nested_event] Caught exception: {err}")
-            logger.debug(f"Exception while handling object: {event_object}")
             raise
 
     def parse_pydantic_event(self, event_object: dict) -> PolymarketEvent:
@@ -67,8 +59,6 @@ def parse_pydantic_event(self, event_object: dict) -> PolymarketEvent:
                 event_object["tags"] = tags
             return PolymarketEvent(**event_object)
         except Exception as err:
-            logger.error(f"[parse_pydantic_event] Caught exception: {err}")
-            logger.debug(f"Exception while handling object: {event_object}")
             raise
 
     def get_markets(
@@ -93,7 +83,6 @@ def get_markets(
                     markets.append(self.parse_pydantic_market(market_object))
                 return markets
         else:
-            logger.error(f"Error response returned from API: HTTP {response.status_code}")
             response.raise_for_status()
             raise RuntimeError(f"API request failed with status {response.status_code}")
 
@@ -119,7 +108,6 @@ def get_events(
                     events.append(self.parse_pydantic_event(market_event_obj))
                 return events
         else:
-            logger.error(f"Error response returned from API: HTTP {response.status_code}")
             response.raise_for_status()
             raise RuntimeError(f"API request failed with status {response.status_code}")
 
@@ -182,7 +170,6 @@ def get_clob_tradable_markets(self, limit=2) -> "list[Market]":
 
     def get_market(self, market_id: int) -> dict:
         url = f"{self.gamma_markets_endpoint}/{market_id}"
-        logger.debug(f"Fetching market from: {url}")
         response = httpx.get(url)
         response.raise_for_status()
         return response.json()