diff --git a/prophet/app.py b/prophet/app.py index f29c9ae..77ec4d7 100644 --- a/prophet/app.py +++ b/prophet/app.py @@ -13,14 +13,14 @@ from prophet.domain.improvement_repo import IImprovementRepo from prophet.domain.original import Original from prophet.infra.improvement_pickle_repo import ImprovementPickleRepo from prophet.infra.improvement_supa_repo import ImprovementSupaRepo -from prophet.llm import LLMClient +from prophet.llm import GroqClient BEE_FEED = "https://babylonbee.com/feed" BEE_FEED_TEST = "test/resources/feed_short.atom" # NOTE: Switch out when done testing REFRESH_PERIOD = 3600 # between fetching articles, in seconds -llm: LLMClient = LLMClient() +llm: GroqClient = GroqClient() repo: IImprovementRepo = ImprovementSupaRepo() diff --git a/prophet/llm.py b/prophet/llm.py index cbe3975..30af7be 100644 --- a/prophet/llm.py +++ b/prophet/llm.py @@ -4,7 +4,7 @@ from prophet.config import AiConfig from prophet.domain.original import Original -class LLMClient: +class GroqClient: config_ai: AiConfig client: Groq