Boris commited on
Commit
b719109
·
1 Parent(s): 8932b1c
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(os.path.dirname(__file__), "service.log")
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(os.path.dirname(__file__), "gigachat.log")
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(os.path.dirname(__file__), "service.log")
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(