Refactor investor and company schemas to allow optional fields; update filtering logic in read_companies function and add find_similar_investors endpoint; change LLM model in InvestorProcessor and QueryProcessor for improved performance.

This commit is contained in:
bolade
2025-09-27 10:45:08 +01:00
parent d36367fbe9
commit 6d902345c0
10 changed files with 234 additions and 46 deletions
+2 -2
View File
@@ -21,8 +21,8 @@ class QueryProcessor:
self.llm = ChatOpenAI(
api_key=os.getenv("OPENROUTER_API_KEY"),
base_url="https://openrouter.ai/api/v1",
model="openai/gpt-5-nano",
temperature=0.3,
model="openai/gpt-4o-mini",
temperature=0,
)
self.toolkit = SQLDatabaseToolkit(db=db, llm=self.llm)
# Update system message to specifically request only investor IDs