Refactor investor-related schemas and models; update database configuration and enhance investor processing logic
This commit is contained in:
@@ -6,7 +6,7 @@ from langchain_community.agent_toolkits import SQLDatabaseToolkit
|
||||
from langchain_community.utilities import SQLDatabase
|
||||
from langchain_openai import ChatOpenAI
|
||||
from langgraph.prebuilt import create_react_agent
|
||||
from pydantic_schemas import Investor, InvestorList
|
||||
from py_schemas import InvestorList
|
||||
from settings import settings
|
||||
|
||||
# Connect to SQLite
|
||||
@@ -29,7 +29,7 @@ class QueryProcessor:
|
||||
api_key=settings.OPENROUTER_API_KEY,
|
||||
base_url="https://openrouter.ai/api/v1",
|
||||
model="google/gemini-2.5-flash-lite",
|
||||
temperature=0,
|
||||
temperature=0.3,
|
||||
)
|
||||
self.toolkit = SQLDatabaseToolkit(db=db, llm=self.llm)
|
||||
self.agent = create_react_agent(
|
||||
|
||||
Reference in New Issue
Block a user