File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11from .request_context import request_context
22from .request_state import RequestState
33from .auth_info import AuthInfo
4- from .follia_module import FolliaModule
4+ from .follia_module import FolliaModule
5+ from .qa_info import QaInfo
Original file line number Diff line number Diff line change 1+ from dataclasses import dataclass
2+
3+ @dataclass
4+ class QaInfo :
5+ """
6+ Data class to hold QA information.
7+ """
8+ agent_id : str = None
9+ session_id : str = None
10+ datasource_id : str = None
11+ interaction_id : str = None
Original file line number Diff line number Diff line change 44
55from zcs .core .session .auth_info import AuthInfo
66from zcs .core .session .follia_module import FolliaModule
7+ from zcs .core .session .qa_info import QaInfo
78
89
910class RequestState ():
@@ -14,14 +15,30 @@ def __init__(
1415 prefix : Optional [str ] = None ,
1516 op_code : Optional [str ] = None ,
1617 auth_info : Optional [AuthInfo ] = None ,
17- follia_module : Optional [FolliaModule ] = None ):
18+ follia_module : Optional [FolliaModule ] = None ,
19+ qa_info : Optional [QaInfo ] = None ):
1820
1921 self .__request_id = request_id if request_id else RequestState .generate_op_code (prefix = prefix )
2022 self .__op_code = op_code if op_code else self .__request_id
2123 self .__request_start_ns = time .perf_counter_ns ()
2224 self .__checkpoint_ns = self .__request_start_ns
2325 self .__auth_info = auth_info
2426 self .__follia_module = follia_module
27+ self .__qa_info = qa_info
28+
29+ def getQaInfo (self ) -> Optional [QaInfo ]:
30+ """
31+ Get QA information.
32+ """
33+
34+ return self .__qa_info
35+
36+ def setQaInfo (self , qa_info : QaInfo ):
37+ """
38+ Set QA information.
39+ """
40+
41+ self .__qa_info = qa_info
2542
2643 def getFolliaModule (self ) -> Optional [FolliaModule ]:
2744 """
You can’t perform that action at this time.
0 commit comments