diff --git a/.env.example b/.env.example
index a5348aa..fa21f7f 100644
--- a/.env.example
+++ b/.env.example
@@ -5,6 +5,7 @@ export AWS_SECRET_ACCESS_KEY=''
 export SOLANA_RPC_URL=''
 export WHITELIST_API_KEY=''
 export OPENROUTER_API_KEY=''
+export GEMINI_API_KEY=''
 export DD_API_KEY=''
 export DD_APP_KEY=''
 export DD_HOSTNAME=''
diff --git a/agent/agent_executors.py b/agent/agent_executors.py
index e9f17bd..5a1c9c6 100644
--- a/agent/agent_executors.py
+++ b/agent/agent_executors.py
@@ -41,8 +41,8 @@
     SUGGESTIONS_MODEL = GOOGLE_GEMINI_20_FLASH_MODEL
     ROUTING_MODEL = GOOGLE_GEMINI_FLASH_15_8B_MODEL
     REASONING_MODEL = GOOGLE_GEMINI_20_FLASH_MODEL
-    BASE_URL = "https://openrouter.ai/api/v1"
-    API_KEY = os.getenv("OPENROUTER_API_KEY")
+    BASE_URL = "https://generativelanguage.googleapis.com/v1beta/"
+    API_KEY = os.getenv("GEMINI_API_KEY")
 else:
     SUGGESTIONS_MODEL = LOCAL_LLM_MODEL
     ROUTING_MODEL = LOCAL_LLM_MODEL
diff --git a/agent/tools.py b/agent/tools.py
index eb1cbfb..8a44699 100644
--- a/agent/tools.py
+++ b/agent/tools.py
@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from langgraph.graph.graph import RunnableConfig
+from langchain_core.runnables.config import RunnableConfig
 from langchain_core.tools import BaseTool, tool
 from agent.telemetry import track_tool_usage
 
diff --git a/onchain/analytics/analytics_tools.py b/onchain/analytics/analytics_tools.py
index 10ca06c..fb0682a 100644
--- a/onchain/analytics/analytics_tools.py
+++ b/onchain/analytics/analytics_tools.py
@@ -1,6 +1,6 @@
 from typing import Dict, Any, List, Tuple, Optional
 from langchain_core.tools import tool
-from langgraph.graph.graph import RunnableConfig
+from langchain_core.runnables.config import RunnableConfig
 import traceback
 import numpy as np
 from enum import StrEnum
diff --git a/onchain/analytics/test_analytics_tools.py b/onchain/analytics/test_analytics_tools.py
index 51db111..17d189f 100644
--- a/onchain/analytics/test_analytics_tools.py
+++ b/onchain/analytics/test_analytics_tools.py
@@ -1,5 +1,5 @@
 import unittest
-from langgraph.graph.graph import RunnableConfig
+from langchain_core.runnables.config import RunnableConfig
 
 from onchain.analytics.analytics_tools import (
     get_coingecko_price_data,
diff --git a/onchain/tokens/trending.py b/onchain/tokens/trending.py
index f252fbc..f4aeeaa 100644
--- a/onchain/tokens/trending.py
+++ b/onchain/tokens/trending.py
@@ -6,7 +6,7 @@
 from typing import Tuple, Optional
 
 from langchain_core.tools import BaseTool, tool
-from langgraph.graph.graph import RunnableConfig
+from langchain_core.runnables.config import RunnableConfig
 from agent.telemetry import track_tool_usage
 
 from api.api_types import TokenMetadata
diff --git a/server/server.py b/server/server.py
index 00d2ad1..f6df647 100644
--- a/server/server.py
+++ b/server/server.py
@@ -1,12 +1,13 @@
-from typing import List, Any, Tuple, Dict, Set
+from typing import List, Any, Tuple, Dict
 import os
 import boto3.dynamodb
 import boto3.dynamodb.table
 import boto3.dynamodb.types
-from flask import Flask, request, jsonify, send_from_directory, g
+from flask import Flask, request, jsonify, send_from_directory
 from flask_cors import CORS
-from pydantic import ValidationError, BaseModel
-from langgraph.graph.graph import CompiledGraph, RunnableConfig
+from pydantic import ValidationError
+from langgraph.graph.graph import CompiledGraph
+from langchain_core.runnables.config import RunnableConfig
 import json
 import traceback
 import boto3