Only firefunction is function calling (#11363)

llama-index-integrations/llms/llama-index-llms-fireworks/llama_index/llms/fireworks/base.py

@@ -7,6 +7,7 @@
 from llama_index.core.types import BaseOutputParser, PydanticProgramMode
 from llama_index.llms.fireworks.utils import (
     fireworks_modelname_to_contextsize,
+    is_function_calling_model,
 )
 from llama_index.llms.openai import OpenAI
 
@@ -64,7 +65,9 @@ def metadata(self) -> LLMMetadata:
             num_output=self.max_tokens,
             is_chat_model=True,
             model_name=self.model,
-            is_function_calling_model=True,
+            is_function_calling_model=is_function_calling_model(
+                model=self._get_model_name()
+            ),
         )
 
     @property

llama-index-integrations/llms/llama-index-llms-fireworks/llama_index/llms/fireworks/utils.py

@@ -63,6 +63,10 @@ def fireworks_modelname_to_contextsize(modelname: str) -> int:
     return context_size
 
 
+def is_function_calling_model(model: str) -> bool:
+    return "function" in model
+
+
 def _message_to_fireworks_prompt(message: ChatMessage) -> Dict[str, Any]:
     if message.role == MessageRole.USER:
         prompt = {"role": "user", "content": message.content}

llama-index-integrations/llms/llama-index-llms-fireworks/pyproject.toml

@@ -26,7 +26,7 @@ description = "llama-index llms fireworks integration"
 license = "MIT"
 name = "llama-index-llms-fireworks"
 readme = "README.md"
-version = "0.1.2"
+version = "0.1.3"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<3.12"