Spaces:
Sleeping
Sleeping
Boris
commited on
Commit
·
b719109
1
Parent(s):
8932b1c
log path
Browse files- src/base_graph.py +0 -10
- src/base_service.py +1 -1
- src/get_answer_gigachat.py +1 -1
- src/get_classification.py +1 -1
src/base_graph.py
CHANGED
@@ -16,7 +16,6 @@ class State(TypedDict):
|
|
16 |
class BaseGraph:
|
17 |
def __init__(self, service: Type[BaseService]):
|
18 |
self.service = service
|
19 |
-
# self._initialize_logging()
|
20 |
self.tools_dict = {tool.name: tool for tool in service.tools}
|
21 |
self.llm_with_tools = AnswerGigaChat().bind_tools(service.tools)
|
22 |
self.messages = service.get_initial_messages()
|
@@ -31,15 +30,6 @@ class BaseGraph:
|
|
31 |
self.messages = service.get_messages_from_redirect(self.messages)
|
32 |
logger.info(f"BaseGraph was rebuilt with service {service}")
|
33 |
|
34 |
-
def _initialize_logging(self):
|
35 |
-
if os.path.exists(self.service.log_path):
|
36 |
-
os.remove(self.service.log_path)
|
37 |
-
logging.basicConfig(
|
38 |
-
level=logging.INFO,
|
39 |
-
format='%(asctime)s - %(levelname)s - %(message)s',
|
40 |
-
handlers=[logging.FileHandler(self.service.log_path)]
|
41 |
-
)
|
42 |
-
|
43 |
def _agent_node(self, state: State):
|
44 |
try:
|
45 |
logger.info("Starting agent_node")
|
|
|
16 |
class BaseGraph:
|
17 |
def __init__(self, service: Type[BaseService]):
|
18 |
self.service = service
|
|
|
19 |
self.tools_dict = {tool.name: tool for tool in service.tools}
|
20 |
self.llm_with_tools = AnswerGigaChat().bind_tools(service.tools)
|
21 |
self.messages = service.get_initial_messages()
|
|
|
30 |
self.messages = service.get_messages_from_redirect(self.messages)
|
31 |
logger.info(f"BaseGraph was rebuilt with service {service}")
|
32 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
def _agent_node(self, state: State):
|
34 |
try:
|
35 |
logger.info("Starting agent_node")
|
src/base_service.py
CHANGED
@@ -33,7 +33,7 @@ class BaseService(ABC):
|
|
33 |
@property
|
34 |
def log_path(self) -> str:
|
35 |
"""Возвращает путь до папки с логами"""
|
36 |
-
return os.path.join(
|
37 |
|
38 |
def get_initial_messages(self):
|
39 |
"""Возвращает начальные сообщения для графа"""
|
|
|
33 |
@property
|
34 |
def log_path(self) -> str:
|
35 |
"""Возвращает путь до папки с логами"""
|
36 |
+
return os.path.join("/tmp", "service.log")
|
37 |
|
38 |
def get_initial_messages(self):
|
39 |
"""Возвращает начальные сообщения для графа"""
|
src/get_answer_gigachat.py
CHANGED
@@ -17,7 +17,7 @@ class AnswerGigaChat(GigaChat):
|
|
17 |
ERROR_MESSAGE_EXCEEDED_COST: str = "\n\n\n\n\nALERT!!!!!!\nCOST NORM WAS EXCEEDED!!!!!!!!\n{} >= " + str(THRESHOLD_COST) + "\n\n\n\n"
|
18 |
COST_PER_INPUT_TOKEN: float = 0.0
|
19 |
COST_PER_OUTPUT_TOKEN: float = 1.95e-3
|
20 |
-
LOG_FILE_PATH: str = os.path.join(
|
21 |
|
22 |
logger: Any = Field(default=None)
|
23 |
|
|
|
17 |
ERROR_MESSAGE_EXCEEDED_COST: str = "\n\n\n\n\nALERT!!!!!!\nCOST NORM WAS EXCEEDED!!!!!!!!\n{} >= " + str(THRESHOLD_COST) + "\n\n\n\n"
|
18 |
COST_PER_INPUT_TOKEN: float = 0.0
|
19 |
COST_PER_OUTPUT_TOKEN: float = 1.95e-3
|
20 |
+
LOG_FILE_PATH: str = os.path.join("/tmp", "gigachat.log")
|
21 |
|
22 |
logger: Any = Field(default=None)
|
23 |
|
src/get_classification.py
CHANGED
@@ -11,7 +11,7 @@ from dotenv import load_dotenv
|
|
11 |
|
12 |
load_dotenv()
|
13 |
|
14 |
-
log_path = os.path.join(
|
15 |
if os.path.exists(log_path):
|
16 |
os.remove(log_path)
|
17 |
logging.basicConfig(
|
|
|
11 |
|
12 |
load_dotenv()
|
13 |
|
14 |
+
log_path = os.path.join("/tmp", "service.log")
|
15 |
if os.path.exists(log_path):
|
16 |
os.remove(log_path)
|
17 |
logging.basicConfig(
|