directory restructure for chroma rag

Polymarket · Jul 31, 2024 · a3f9485 · a3f9485
1 parent 7f18cbd
commit a3f9485
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 10 deletions.
diff --git a/ai/rag/polymarket_rag.py → ai/rag/chroma.py b/ai/rag/polymarket_rag.py → ai/rag/chroma.py
diff --git a/api/polymarket/gamma.py → api/gamma.py b/api/polymarket/gamma.py → api/gamma.py
@@ -1,11 +1,10 @@
-import sys
 import httpx
 import json
 
-from api.polymarket.types import PolymarketEvent
-from api.polymarket.types import Market
-from api.polymarket.types import ClobReward
-from api.polymarket.types import Tag
+from api.types import PolymarketEvent
+from api.types import Market
+from api.types import ClobReward
+from api.types import Tag
 
 
 class GammaMarketClient:

diff --git a/api/polymarket/polymarket.py → api/polymarket.py b/api/polymarket/polymarket.py → api/polymarket.py
diff --git a/api/polymarket/__init__.py b/api/polymarket/__init__.py
diff --git a/api/polymarket/types.py → api/types.py b/api/polymarket/types.py → api/types.py
diff --git a/scripts/python/cli.py b/scripts/python/cli.py
@@ -1,15 +1,12 @@
 import typer
-import pdb
 from devtools import pprint
 
-from api.polymarket.polymarket import Polymarket
+from api.polymarket import Polymarket
 from ai.llm import executor, prompts
-from ai.rag.polymarket_rag import PolymarketRAG
+from ai.rag.chroma import PolymarketRAG
 from data.news_providers.newsapi_org.newsapi_caller import NewsApiCaller
 from jobs.scheduler import TradingAgent
 
-from langchain_core.output_parsers import StrOutputParser
-
 app = typer.Typer()
 polymarket = Polymarket()
 newsapi_client = NewsApiCaller()