mirror of
https://github.com/johndoe6345789/AutoMetabuilder.git
synced 2026-04-26 14:54:57 +00:00
23 lines
718 B
Python
23 lines
718 B
Python
"""Build workflow runtime context."""
|
|
import os
|
|
|
|
DEFAULT_MODEL = "openai/gpt-4o"
|
|
|
|
|
|
def resolve_model_name(prompt: dict) -> str:
|
|
"""Resolve model name from env or prompt."""
|
|
return os.environ.get("LLM_MODEL", prompt.get("model", DEFAULT_MODEL))
|
|
|
|
|
|
def build_workflow_context(parts: dict) -> dict:
|
|
"""Build the workflow context dict."""
|
|
context = dict(parts)
|
|
# Only resolve model if prompt is available, otherwise use default
|
|
if "prompt" in parts:
|
|
prompt = parts["prompt"]
|
|
context["model_name"] = resolve_model_name(prompt)
|
|
else:
|
|
# Workflow plugins will load prompt, model will be resolved then
|
|
context["model_name"] = resolve_model_name({})
|
|
return context
|