File size: 871 Bytes
1094f14
96f6b21
 
1094f14
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from langchain_core.tools import BaseTool, tool
from base_service import BaseService
from template._prompts import SYSTEM_PROMPT ## FIX
from typing import List
import logging
import time

logger = logging.getLogger(__name__)


class ExamplePLSRenameService(BaseService):
    @property
    def system_prompt(self) -> str:
        return SYSTEM_PROMPT
    
    @property
    def tools(self) -> List[BaseTool]:
        return self._base_tools + [
            self.example_func,
        ]

    @tool
    def example_func(question: str) -> None:
        """Пример функции, как её здесь оформлять. Просто для примера, нужно удалить при реальном использовании"""
        print("Я example_func")
        time.sleep(4)
        logger.info("Кто-то забыл удалить example_func!")