diff --git a/llama-index-integrations/llms/llama-index-llms-groq/llama_index/llms/groq/base.py b/llama-index-integrations/llms/llama-index-llms-groq/llama_index/llms/groq/base.py index 98155671e5fe3299cff0272138c0e7fef4b1c48d..c88c4b29fba31497b80cb473fa18f6dfcdf87aa8 100644 --- a/llama-index-integrations/llms/llama-index-llms-groq/llama_index/llms/groq/base.py +++ b/llama-index-integrations/llms/llama-index-llms-groq/llama_index/llms/groq/base.py @@ -29,6 +29,7 @@ class Groq(OpenAILike): api_key: Optional[str] = None, api_base: str = "https://api.groq.com/openai/v1", is_chat_model: bool = True, + is_function_calling_model: bool = True, **kwargs: Any, ) -> None: api_key = api_key or os.environ.get("GROQ_API_KEY", None) @@ -37,6 +38,7 @@ class Groq(OpenAILike): api_key=api_key, api_base=api_base, is_chat_model=is_chat_model, + is_function_calling_model=is_function_calling_model, **kwargs, ) diff --git a/llama-index-integrations/llms/llama-index-llms-groq/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-groq/pyproject.toml index 0a90d93ead5e57eacebe48fb61085b4217a2c4b6..45d068a8491c3b66cd07a5bdc254c6c006d3ca8d 100644 --- a/llama-index-integrations/llms/llama-index-llms-groq/pyproject.toml +++ b/llama-index-integrations/llms/llama-index-llms-groq/pyproject.toml @@ -27,7 +27,7 @@ exclude = ["**/BUILD"] license = "MIT" name = "llama-index-llms-groq" readme = "README.md" -version = "0.1.3" +version = "0.1.4" [tool.poetry.dependencies] python = ">=3.8.1,<4.0"