diff --git a/packages/ragbits-core/src/ragbits/core/llms/base.py b/packages/ragbits-core/src/ragbits/core/llms/base.py index ce74d07ee..3b37914ca 100644 --- a/packages/ragbits-core/src/ragbits/core/llms/base.py +++ b/packages/ragbits-core/src/ragbits/core/llms/base.py @@ -80,7 +80,7 @@ async def generate_raw( Raw text response from LLM. """ merged_options = (self.default_options | options) if options else self.default_options - response = await self.client.call( + response = await self.call( conversation=self._format_chat_for_llm(prompt), options=merged_options, json_mode=prompt.json_mode, @@ -146,7 +146,7 @@ async def generate_streaming( Response stream from LLM. """ merged_options = (self.default_options | options) if options else self.default_options - response = await self.client.call_streaming( + response = await self.call_streaming( conversation=self._format_chat_for_llm(prompt), options=merged_options, json_mode=prompt.json_mode,