mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-10 05:19:44 +00:00
feat(awel): AWEL supports http trigger
This commit is contained in:
54
examples/awel/simple_chat_dag_example.py
Normal file
54
examples/awel/simple_chat_dag_example.py
Normal file
@@ -0,0 +1,54 @@
|
||||
"""AWEL: Simple chat dag example
|
||||
|
||||
Example:
|
||||
|
||||
.. code-block:: shell
|
||||
|
||||
curl -X POST http://127.0.0.1:5000/api/v1/awel/trigger/examples/simple_chat \
|
||||
-H "Content-Type: application/json" -d '{
|
||||
"model": "proxyllm",
|
||||
"user_input": "hello"
|
||||
}'
|
||||
"""
|
||||
from typing import Dict
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from pilot.awel import DAG, HttpTrigger, MapOperator
|
||||
from pilot.scene.base_message import ModelMessage
|
||||
from pilot.model.base import ModelOutput
|
||||
from pilot.model.operator.model_operator import ModelOperator
|
||||
|
||||
|
||||
class TriggerReqBody(BaseModel):
|
||||
model: str = Field(..., description="Model name")
|
||||
user_input: str = Field(..., description="User input")
|
||||
|
||||
|
||||
class RequestHandleOperator(MapOperator[TriggerReqBody, Dict]):
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
|
||||
async def map(self, input_value: TriggerReqBody) -> Dict:
|
||||
hist = []
|
||||
hist.append(ModelMessage.build_human_message(input_value.user_input))
|
||||
hist = list(h.dict() for h in hist)
|
||||
params = {
|
||||
"prompt": input_value.user_input,
|
||||
"messages": hist,
|
||||
"model": input_value.model,
|
||||
"echo": False,
|
||||
}
|
||||
print(f"Receive input value: {input_value}")
|
||||
return params
|
||||
|
||||
|
||||
with DAG("dbgpt_awel_simple_dag_example") as dag:
|
||||
# Receive http request and trigger dag to run.
|
||||
trigger = HttpTrigger(
|
||||
"/examples/simple_chat", methods="POST", request_body=TriggerReqBody
|
||||
)
|
||||
request_handle_task = RequestHandleOperator()
|
||||
model_task = ModelOperator()
|
||||
# type(out) == ModelOutput
|
||||
model_parse_task = MapOperator(lambda out: out.to_dict())
|
||||
trigger >> request_handle_task >> model_task >> model_parse_task
|
32
examples/awel/simple_dag_example.py
Normal file
32
examples/awel/simple_dag_example.py
Normal file
@@ -0,0 +1,32 @@
|
||||
"""AWEL: Simple dag example
|
||||
|
||||
Example:
|
||||
|
||||
.. code-block:: shell
|
||||
|
||||
curl -X GET http://127.0.0.1:5000/api/v1/awel/trigger/examples/hello\?name\=zhangsan
|
||||
|
||||
"""
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from pilot.awel import DAG, HttpTrigger, MapOperator
|
||||
|
||||
|
||||
class TriggerReqBody(BaseModel):
|
||||
name: str = Field(..., description="User name")
|
||||
age: int = Field(18, description="User age")
|
||||
|
||||
|
||||
class RequestHandleOperator(MapOperator[TriggerReqBody, str]):
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
|
||||
async def map(self, input_value: TriggerReqBody) -> str:
|
||||
print(f"Receive input value: {input_value}")
|
||||
return f"Hello, {input_value.name}, your age is {input_value.age}"
|
||||
|
||||
|
||||
with DAG("simple_dag_example") as dag:
|
||||
trigger = HttpTrigger("/examples/hello", request_body=TriggerReqBody)
|
||||
map_node = RequestHandleOperator()
|
||||
trigger >> map_node
|
Reference in New Issue
Block a user