Fix runnableconfig

.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=''

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

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
 

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

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,

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

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