Package opengradient.mltools

OpenGradient AlphaSense Tools

Functions

Create og model tool

def create_og_model_tool(tool_type: opengradient.mltools.model_tool.ToolType, model_cid: str, tool_name: str, input_getter: Callable, output_formatter: Callable[..., str], input_schema: Type[pydantic.main.BaseModel] = None, tool_description: str = 'Executes the given ML model', inference_mode: opengradient.types.InferenceMode = 0) ‑> langchain_core.tools.base.BaseTool

Creates a tool that wraps an OpenGradient model for inference.

This function generates a tool that can be integrated into either a LangChain pipeline or a Swarm system, allowing the model to be executed as part of a chain of operations. The tool uses the provided input_getter function to obtain the necessary input data and runs inference using the specified OpenGradient model.

Returns

BaseTool: For ToolType.LANGCHAIN, returns a LangChain StructuredTool. Callable: For ToolType.SWARM, returns a decorated function with appropriate metadata.

Raises

• ValueError: If an invalid tool_type is provided.

Classes

ToolType

class ToolType(*args, **kwds)

Indicates the framework the tool is compatible with.

Variables

• static LANGCHAIN

• static SWARM