From 8edcc63b7e8476558d4d578836bfc3e010223a54 Mon Sep 17 00:00:00 2001
From: Rick Lamers <ricklamers@gmail.com>
Date: Fri, 17 May 2024 22:29:45 +0200
Subject: [PATCH] fix: enable tool use (#13566)
---
.../llms/llama-index-llms-groq/llama_index/llms/groq/base.py | 2 ++
.../llms/llama-index-llms-groq/pyproject.toml | 2 +-
2 files changed, 3 insertions(+), 1 deletion(-)
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 98155671e5..c88c4b29fb 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 0a90d93ead..45d068a849 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"
--
GitLab