Move cost collector away from the llm dir

whiletrue-industries · Jun 8, 2024 · 8c5d5fe · 8c5d5fe
1 parent f91abf8
commit 8c5d5fe
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/odds/common/llm/cost_collector.py → odds/common/cost_collector.py b/odds/common/llm/cost_collector.py → odds/common/cost_collector.py
diff --git a/odds/common/embedder/openai/openai_embedder.py b/odds/common/embedder/openai/openai_embedder.py
@@ -5,7 +5,7 @@
 from ...datatypes import Embedding
 
 from ..embedder import Embedder
-from ...llm.cost_collector import CostCollector
+from ...cost_collector import CostCollector
 from ...config import config
 from ...retry import Retry
 

diff --git a/odds/common/llm/llm_runner.py b/odds/common/llm/llm_runner.py
@@ -1,6 +1,6 @@
 from .llm_query import LLMQuery
 from .llm_cache import LLMCache
-from .cost_collector import CostCollector
+from ..cost_collector import CostCollector
 
 class LLMRunner:
 

diff --git a/test_assistant.py b/test_assistant.py
@@ -7,7 +7,7 @@
 from odds.common.store import store
 from odds.common.embedder import embedder
 from odds.common.catalog_repo import catalog_repo
-from odds.common.llm.cost_collector import CostCollector
+from odds.common.cost_collector import CostCollector
 from odds.common.llm.openai.openai_llm_runner import OpenAILLMRunner
 import sqlite3