adding tests and imports to inits

haystack_experimental/components/__init__.py

@@ -1,3 +1,7 @@
 # SPDX-FileCopyrightText: 2022-present deepset GmbH <[email protected]>
 #
 # SPDX-License-Identifier: Apache-2.0
+
+from .tools import OpenAIFunctionCaller
+
+_all_ = [ "OpenAIFunctionCaller"]

haystack_experimental/components/tools/__init__.py

@@ -1,3 +1,7 @@
 # SPDX-FileCopyrightText: 2022-present deepset GmbH <[email protected]>
 #
 # SPDX-License-Identifier: Apache-2.0
+
+from .openai.function_caller import OpenAIFunctionCaller
+
+_all_ = ["OpenAIFunctionCaller"]

haystack_experimental/components/tools/openai/__init__.py

@@ -1,3 +1,7 @@
 # SPDX-FileCopyrightText: 2022-present deepset GmbH <[email protected]>
 #
 # SPDX-License-Identifier: Apache-2.0
+
+from .function_caller import OpenAIFunctionCaller
+
+_all_ = [ "OpenAIFunctionCaller"]

haystack_experimental/components/tools/openai/function_caller.py

@@ -5,7 +5,7 @@
 import json
 from typing import Any, Callable, Dict, List
 
-from haystack import component, default_to_dict
+from haystack import component, default_from_dict, default_to_dict
 from haystack.dataclasses import ChatMessage
 from haystack.utils import deserialize_callable, serialize_callable
 
@@ -62,7 +62,7 @@ def from_dict(cls, data: Dict[str, Any]) -> "OpenAIFunctionCaller":
         for name, path in available_function_paths.items():
             available_functions[name] = deserialize_callable(path)
         data["init_parameters"]["available_functions"] = available_functions
-        return default_to_dict(cls, data)
+        return default_from_dict(cls, data)
 
     @component.output_types(function_replies=List[ChatMessage], assistant_replies=List[ChatMessage])
     def run(self, messages: List[ChatMessage]):

test/components/__init__.py

@@ -0,0 +1,3 @@
+# SPDX-FileCopyrightText: 2022-present deepset GmbH <[email protected]>
+#
+# SPDX-License-Identifier: Apache-2.0

test/components/tools/__init__.py

@@ -0,0 +1,3 @@
+# SPDX-FileCopyrightText: 2022-present deepset GmbH <[email protected]>
+#
+# SPDX-License-Identifier: Apache-2.0

test/components/tools/openai/__init__.py

@@ -0,0 +1,3 @@
+# SPDX-FileCopyrightText: 2022-present deepset GmbH <[email protected]>
+#
+# SPDX-License-Identifier: Apache-2.0

test/components/tools/openai/test_function_caller.py

@@ -0,0 +1,69 @@
+# SPDX-FileCopyrightText: 2023-present deepset GmbH <[email protected]>
+#
+# SPDX-License-Identifier: Apache-2.0
+import os
+import json
+import pytest
+
+# from haystack.utils import Secret
+from haystack_experimental.components.tools import OpenAIFunctionCaller
+from haystack.dataclasses import ChatMessage
+
+WEATHER_INFO = {
+    "Berlin": {"weather": "mostly sunny", "temperature": 7, "unit": "celsius"},
+    "Paris": {"weather": "mostly cloudy", "temperature": 8, "unit": "celsius"},
+    "Rome": {"weather": "sunny", "temperature": 14, "unit": "celsius"},
+    "Madrid": {"weather": "sunny", "temperature": 10, "unit": "celsius"},
+    "London": {"weather": "cloudy", "temperature": 9, "unit": "celsius"},
+}
+
+
+def mock_weather_func(location):
+    if location in WEATHER_INFO:
+        return WEATHER_INFO[location]
+    else:
+        return {"weather": "sunny", "temperature": 21.8, "unit": "fahrenheit"}
+
+class TestOpenAIFunctionCaller:
+
+    def test_init(self, monkeypatch):
+        component = OpenAIFunctionCaller(available_functions = {"mock_weather_func": mock_weather_func})
+        assert component.available_functions == {"mock_weather_func": mock_weather_func}
+
+    def test_successful_function_call(self, monkeypatch):
+        component = OpenAIFunctionCaller(available_functions = {"mock_weather_func": mock_weather_func})
+        mock_assistant_message = ChatMessage.from_assistant(content='[{"id": "mock-id", "function": {"arguments": "{\\"location\\":\\"Berlin\\"}", "name": "mock_weather_func"}, "type": "function"}]',
+                                                                meta={"finish_reason": "tool_calls"})
+        result = component.run(messages=[mock_assistant_message])
+        result_obj = json.loads(result["function_replies"][-1].content)
+        assert result_obj['weather'] == WEATHER_INFO['Berlin']['weather']
+        assert result_obj['temperature'] == WEATHER_INFO['Berlin']['temperature']
+        assert result_obj['unit'] == WEATHER_INFO['Berlin']['unit']
+
+
+    def test_failing_function_call(self, monkeypatch):
+        component = OpenAIFunctionCaller(available_functions = {"mock_weather_func": mock_weather_func})
+        mock_assistant_message = ChatMessage.from_assistant(content='[{"id": "mock-id", "function": {"arguments": "{\\"location\\":\\"Berlin\\"}", "name": "mock_weather"}, "type": "function"}]',
+                                                                meta={"finish_reason": "tool_calls"})
+        result = component.run(messages=[mock_assistant_message])
+        assert result["function_replies"][-1].content == "I'm sorry, I tried to run a function that did not exist. Would you like me to correct it and try again?"
+
+    def test_to_dict(self, monkeypatch):
+        component = OpenAIFunctionCaller(available_functions = {"mock_weather_func": mock_weather_func})
+        data = component.to_dict()
+        assert data == {
+            "type": "haystack_experimental.components.tools.openai.function_caller.OpenAIFunctionCaller",
+            "init_parameters": {
+                "available_functions": {'mock_weather_func': 'test.components.tools.openai.test_function_caller.mock_weather_func'}
+            },
+        }
+
+    def test_from_dict(self, monkeypatch):
+        data = {
+            "type": "haystack_experimental.components.tools.openai.function_caller.OpenAIFunctionCaller",
+            "init_parameters": {
+                "available_functions": {'mock_weather_func': 'test.components.tools.openai.test_function_caller.mock_weather_func'},
+            },
+        }
+        component: OpenAIFunctionCaller = OpenAIFunctionCaller.from_dict(data)
+        assert component.available_functions == {'mock_weather_func': mock_weather_func}