feat(projekt-matching): seven Langflow wrapper components
This commit is contained in:
33
projekt-matching/components/pm_match.py
Normal file
33
projekt-matching/components/pm_match.py
Normal file
@@ -0,0 +1,33 @@
|
||||
import sys
|
||||
|
||||
if "/app/langflow" not in sys.path:
|
||||
sys.path.insert(0, "/app/langflow")
|
||||
|
||||
from langflow.custom import Component
|
||||
from langflow.io import DataInput, MessageTextInput, Output
|
||||
from langflow.schema import Data
|
||||
|
||||
from projektmatch import stages
|
||||
from projektmatch.config import Cfg
|
||||
|
||||
|
||||
class PMMatch(Component):
|
||||
display_name = "PM 3 Match CV"
|
||||
description = "LLM 2: Anforderungen gegen Lebenslauf bewerten"
|
||||
inputs = [
|
||||
DataInput(name="ctx", display_name="Context"),
|
||||
MessageTextInput(name="vllm_base", display_name="vLLM Base",
|
||||
value="PM_VLLM_BASE", load_from_db=True,
|
||||
advanced=True),
|
||||
MessageTextInput(name="vllm_model", display_name="vLLM Model",
|
||||
value="PM_VLLM_MODEL", load_from_db=True,
|
||||
advanced=True),
|
||||
]
|
||||
outputs = [Output(name="out", display_name="Context", method="build_out")]
|
||||
|
||||
def build_out(self) -> Data:
|
||||
cfg = Cfg(vllm_base=self.vllm_base, vllm_model=self.vllm_model)
|
||||
ctx = stages.run_stage("match", stages.stage_match,
|
||||
dict(self.ctx.data), cfg)
|
||||
self.status = ctx.get("status", "")
|
||||
return Data(data=ctx)
|
||||
Reference in New Issue
Block a user