wehub-resource-sync
e768098d0e
tools_continuous_delivery / Private PyPI non-main branch release (push) Has been skipped
tools_continuous_delivery / Private PyPI main branch release (push) Failing after 2m42s
Publish Promptflow Doc / Build (push) Has been cancelled
Publish Promptflow Doc / Deploy (push) Has been cancelled
Flake8 Lint / flake8 (push) Has been cancelled
Spell check CI / Spell_Check (push) Has been cancelled
20 lines
424 B
Python
20 lines
424 B
Python
import os
|
|
|
|
from promptflow.core import tool
|
|
from promptflow.connections import CustomConnection
|
|
|
|
from intent import extract_intent
|
|
|
|
|
|
@tool
|
|
def extract_intent_tool(chat_prompt, connection: CustomConnection) -> str:
|
|
|
|
# set environment variables
|
|
for key, value in dict(connection).items():
|
|
os.environ[key] = value
|
|
|
|
# call the entry function
|
|
return extract_intent(
|
|
chat_prompt=chat_prompt,
|
|
)
|