From a3f9485e083b50e6da1501a2ed91c4b3fa9b9499 Mon Sep 17 00:00:00 2001
From: alexnicita <alexandernicita@gmail.com>
Date: Wed, 31 Jul 2024 17:39:36 -0400
Subject: [PATCH] directory restructure for chroma rag

---
 ai/rag/{polymarket_rag.py => chroma.py} | 0
 api/{polymarket => }/gamma.py           | 9 ++++-----
 api/{polymarket => }/polymarket.py      | 0
 api/polymarket/__init__.py              | 0
 api/{polymarket => }/types.py           | 0
 scripts/python/cli.py                   | 7 ++-----
 6 files changed, 6 insertions(+), 10 deletions(-)
 rename ai/rag/{polymarket_rag.py => chroma.py} (100%)
 rename api/{polymarket => }/gamma.py (97%)
 rename api/{polymarket => }/polymarket.py (100%)
 delete mode 100644 api/polymarket/__init__.py
 rename api/{polymarket => }/types.py (100%)

diff --git a/ai/rag/polymarket_rag.py b/ai/rag/chroma.py
similarity index 100%
rename from ai/rag/polymarket_rag.py
rename to ai/rag/chroma.py
diff --git a/api/polymarket/gamma.py b/api/gamma.py
similarity index 97%
rename from api/polymarket/gamma.py
rename to api/gamma.py
index 18af921..29fa003 100644
--- a/api/polymarket/gamma.py
+++ b/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 b/api/polymarket.py
similarity index 100%
rename from api/polymarket/polymarket.py
rename to api/polymarket.py
diff --git a/api/polymarket/__init__.py b/api/polymarket/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/api/polymarket/types.py b/api/types.py
similarity index 100%
rename from api/polymarket/types.py
rename to api/types.py
diff --git a/scripts/python/cli.py b/scripts/python/cli.py
index 772c3d2..3c9e48d 100644
--- 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()