Skip to content

Commit 10ce223

Browse files
authored
Merge pull request #14 from zcsadmin/13-modifica-zcs-core-per-aggiungere-qa_info-refs-ap-2094
add qa_info
2 parents 5be26dd + 0ee0158 commit 10ce223

3 files changed

Lines changed: 31 additions & 2 deletions

File tree

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from .request_context import request_context
22
from .request_state import RequestState
33
from .auth_info import AuthInfo
4-
from .follia_module import FolliaModule
4+
from .follia_module import FolliaModule
5+
from .qa_info import QaInfo
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
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

app/src/zcs/core/session/request_state.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
from zcs.core.session.auth_info import AuthInfo
66
from zcs.core.session.follia_module import FolliaModule
7+
from zcs.core.session.qa_info import QaInfo
78

89

910
class 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
"""

0 commit comments

Comments
 (0)