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

8
services/extractor.py
View File

@ -4,7 +4,7 @@ from typing import Callable, Type, Any, TypeVar
from langchain_core.runnables import Runnable
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import PydanticOutputParser
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_openai.chat_models import ChatOpenAI
from pydantic import BaseModel
from utils.parsers import process_profile, process_passport, process_account
@ -73,7 +73,7 @@ def extract_passport(client_data: dict[str, Any]) -> FromPassport:
def extract_profile(client_data: dict[str, Any]) -> FromProfile:
passport_data = client_data.get("profile")
profile_data = client_data.get("profile")
prompt_template = (
"Extract the following information from the provided text.\n"
@ -83,7 +83,7 @@ def extract_profile(client_data: dict[str, Any]) -> FromProfile:
)
result = __run_extraction_chain(
raw_file_data=passport_data,
raw_file_data=profile_data,
file_processor=process_profile,
pydantic_model=FromProfile,
prompt_template=prompt_template,
@ -125,7 +125,7 @@ def __run_extraction_chain(
prompt = ChatPromptTemplate.from_template(prompt_template)
chain: Runnable = prompt | ChatGoogleGenerativeAI(model=model_name) | parser
chain: Runnable = prompt | ChatOpenAI(model="gpt-4o-mini") | parser
result = chain.invoke({
"processed_text": processed_text,