dlmw/julius_baer_onboarding
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor using openai

Browse Source
This commit is contained in:
robinrolle
2025-04-13 01:13:20 +02:00
parent a440aee411
commit 537527fba0
3 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
services/advisor.py
View File

@ -7,7 +7,7 @@ from dto.requests import GameStartRequestDTO, GameDecisionRequestDTO
from services.extractor import extract_profile, extract_passport, extract_description, extract_account
from services.julius_baer_api_client import JuliusBaerApiClient
from utils.storage.game_files_manager import store_game_round_data
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_openai.chat_models import ChatOpenAI
from validation.llm_validate import AdvisorDecision
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import PydanticOutputParser
@ -129,7 +129,7 @@ class Advisor:
)
# 4. Chaîne LLM
chain = prompt | ChatGoogleGenerativeAI(model="gemini-2.0-flash") | parser
chain = prompt | ChatOpenAI(model="gpt-4o-mini") | parser
# 5. Invocation
result: AdvisorDecision = chain.invoke({