diff --git a/langchain_opengradient/toolkits.py b/langchain_opengradient/toolkits.py
index 6d06010..a91d3d1 100644
--- a/langchain_opengradient/toolkits.py
+++ b/langchain_opengradient/toolkits.py
@@ -151,7 +151,7 @@ def create_run_model_tool(
         self,
         model_cid: str,
         tool_name: str,
-        model_input_provider: Callable[..., InferenceResult],
+        model_input_provider: Callable[..., dict],
         model_output_formatter: Callable[..., str],
         tool_input_schema: Optional[Type[BaseModel]] = None,
         tool_description: str = "Executes the given ML model",
@@ -215,7 +215,7 @@ def GetInputData():
             eth_volatility_tool = toolkit.create_run_model_tool(
                 model_cid = "QmRhcpDXfYCKsimTmJYrAVM4Bbvck59Zb2onj3MHv9Kw5N",
                 tool_name = "one_hour_eth_usdt_volatility",
-                model_input_provider = GetInputData(),
+                model_input_provider = GetInputData,
                 model_output_formatter = lambda x: format(
                     float(x.model_output["Y"].item()),
                     ".3%"
@@ -276,7 +276,6 @@ def create_read_workflow_tool(
 
             toolkit = OpenGradientToolkit()
             btc_workflow_tool = toolkit.create_read_workflow_tool(
-                tool_type=ToolType.LANGCHAIN,
                 workflow_contract_address="0x6e0641925b845A1ca8aA9a890C4DEF388E9197e0",
                 tool_name="ETH_Price_Forecast",
                 tool_description="Reads latest forecast for ETH price",
diff --git a/pyproject.toml b/pyproject.toml
index f68dfaf..f0165ae 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,15 +8,15 @@ version = "0.1.1"
 description = "An integration package connecting OpenGradient and LangChain"
 authors = []
 readme = "README.md"
-repository = "https://github.com/langchain-ai/langchain"
+repository = "https://github.com/OpenGradient/og-langchain"
 license = "MIT"
 
 [tool.mypy]
 disallow_untyped_defs = "True"
 
 [tool.poetry.urls]
-"Source Code" = "https://github.com/langchain-ai/langchain/tree/master/libs/partners/opengradient"
-"Release Notes" = "https://github.com/langchain-ai/langchain/releases?q=tag%3A%22opengradient%3D%3D0%22&expanded=true"
+"Source Code" = "https://github.com/OpenGradient/og-langchain"
+"Release Notes" = "https://github.com/OpenGradient/og-langchain/releases"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"