-
Notifications
You must be signed in to change notification settings - Fork 18
Expand file tree
/
Copy pathfsm_definition.py
More file actions
42 lines (31 loc) · 1.14 KB
/
fsm_definition.py
File metadata and controls
42 lines (31 loc) · 1.14 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
from chatfaq_sdk import ChatFAQSDK
from chatfaq_sdk.fsm import FSMDefinition, State, Transition
from chatfaq_sdk.layers import Message, StreamingMessage
from chatfaq_sdk.clients import llm_request
from chatfaq_sdk.utils import convert_mml_to_llm_format
async def send_greeting(sdk: ChatFAQSDK, ctx: dict):
yield Message("How can we help you?")
async def send_answer(sdk: ChatFAQSDK, ctx: dict):
messages = convert_mml_to_llm_format(ctx["conv_mml"][1:]) # skip the greeting message
messages.insert(0, {"role": "system", "content": "You are a helpful assistant."})
generator = llm_request(
sdk,
"gpt-4o",
use_conversation_context=True,
conversation_id=ctx["conversation_id"],
bot_channel_name=ctx["bot_channel_name"],
messages=messages,
)
yield StreamingMessage(generator)
greeting_state = State(name="Greeting", events=[send_greeting], initial=True)
answering_state = State(
name="Answering",
events=[send_answer],
)
_to_answer = Transition(
dest=answering_state,
)
fsm_definition = FSMDefinition(
states=[greeting_state, answering_state],
transitions=[_to_answer]
)