■ BaseTool 클래스의 inovoke 메소드를 사용해 컨텐트/아티펙트를 구하는 방법을 보여준다.
▶ main.py
|
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 30 31 32 33 34 35 36 37 38 39 40 |
import random from langchain_core.tools import BaseTool from typing import Tuple from typing import List class GenerateRandomFloatValueTool(BaseTool): name : str = "GenerateRandomFloatValueTool" description : str = "Generate size random floats in the range [minimum, maximum]." response_format : str = "content_and_artifact" digitCount : int = 2 def _run(self, minimum : float, maximum : float, count : int) -> Tuple[str, List[float]]: range_ = maximum - minimum array = [ round(minimum + (range_ * random.random()), ndigits = self.digitCount) for _ in range(count) ] content = f"Generated {count} floats in [{minimum}, {maximum}], rounded to {self.digitCount} decimals." return content, array generateRandomFloatValueTool = GenerateRandomFloatValueTool(digitCount = 4) responseToolMessage = generateRandomFloatValueTool.invoke( { "name" : "generateRandomFloatValueTool", "args" : {"minimum" : 0.1, "maximum" : 3.3333, "count" : 3}, "id" : "123", # required "type" : "tool_call" # required } ) print(responseToolMessage) """ content='Generated 3 floats in [0.1, 3.3333], rounded to 4 decimals.' name='GenerateRandomFloatValueTool' tool_call_id='123' artifact=[1.5382, 2.6612, 2.3324] """ |
▶ requirements.txt
|
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 30 31 32 33 34 35 36 37 38 |
aiohappyeyeballs==2.4.2 aiohttp==3.10.8 aiosignal==1.3.1 annotated-types==0.7.0 anyio==4.6.0 async-timeout==4.0.3 attrs==24.2.0 certifi==2024.8.30 charset-normalizer==3.3.2 exceptiongroup==1.2.2 frozenlist==1.4.1 greenlet==3.1.1 h11==0.14.0 httpcore==1.0.5 httpx==0.27.2 idna==3.10 jsonpatch==1.33 jsonpointer==3.0.0 langchain==0.3.1 langchain-core==0.3.6 langchain-text-splitters==0.3.0 langsmith==0.1.129 multidict==6.1.0 numpy==1.26.4 orjson==3.10.7 packaging==24.1 pydantic==2.9.2 pydantic_core==2.23.4 PyYAML==6.0.2 requests==2.32.3 sniffio==1.3.1 SQLAlchemy==2.0.35 tenacity==8.5.0 typing_extensions==4.12.2 urllib3==2.2.3 yarl==1.13.1 |
※ pip install langchain 명령을 실행했다.