[langchain] fix OpenAIAssistantRunnable.create_assistant

libs/langchain/langchain/agents/openai_assistant/base.py

@@ -3,12 +3,23 @@
 import json
 from json import JSONDecodeError
 from time import sleep
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Sequence, Tuple, Union
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Callable,
+    Dict,
+    List,
+    Optional,
+    Sequence,
+    Tuple,
+    Type,
+    Union,
+)
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.callbacks import CallbackManager
 from langchain_core.load import dumpd
-from langchain_core.pydantic_v1 import Field, root_validator
+from langchain_core.pydantic_v1 import BaseModel, Field, root_validator
 from langchain_core.runnables import RunnableConfig, RunnableSerializable, ensure_config
 from langchain_core.tools import BaseTool
 from langchain_core.utils.function_calling import convert_to_openai_tool
@@ -76,6 +87,20 @@ def _get_openai_async_client() -> openai.AsyncOpenAI:
         ) from e
 
 
+def _get_assistants_tool(
+    tool: Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool],
+) -> Dict[str, Any]:
+    """Convert a raw function/class to an OpenAI tool.
+
+    Note that OpenAI assistants supports several built-in tools,
+    such as "code_interpreter" and "retrieval."
+    """
+    if isinstance(tool, dict) and "type" in tool:
+        return tool
+    else:
+        return convert_to_openai_tool(tool)
+
+
 OutputType = Union[
     List[OpenAIAssistantAction],
     OpenAIAssistantFinish,
@@ -210,7 +235,7 @@ def create_assistant(
         assistant = client.beta.assistants.create(
             name=name,
             instructions=instructions,
-            tools=[convert_to_openai_tool(tool) for tool in tools],  # type: ignore
+            tools=[_get_assistants_tool(tool) for tool in tools],  # type: ignore
             model=model,
         )
         return cls(assistant_id=assistant.id, client=client, **kwargs)

libs/langchain/tests/unit_tests/agents/test_openai_assistant.py

@@ -1,8 +1,17 @@
+from typing import Any
+from unittest.mock import MagicMock, patch
+
 import pytest
 
 from langchain.agents.openai_assistant import OpenAIAssistantRunnable
 
 
+def _create_mock_client(*args: Any, **kwargs: Any) -> Any:
+    client = MagicMock()
+    client.beta.assistants.create().id = "abc123"
+    return client
+
+
 @pytest.mark.requires("openai")
 def test_user_supplied_client() -> None:
     import openai
@@ -19,3 +28,18 @@ def test_user_supplied_client() -> None:
     )
 
     assert assistant.client == client
+
+
+@pytest.mark.requires("openai")
+@patch(
+    "langchain.agents.openai_assistant.base._get_openai_client",
+    new=_create_mock_client,
+)
+def test_create_assistant() -> None:
+    assistant = OpenAIAssistantRunnable.create_assistant(
+        name="name",
+        instructions="instructions",
+        tools=[{"type": "code_interpreter"}],
+        model="",
+    )
+    assert isinstance(assistant, OpenAIAssistantRunnable)