/
hernantos
/
test-agent
Обзор
Документация
Войти
/
hernantos
/
test-agent
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
main.py
28 строк
841 B
hernantos
init
03 июл 2025, 13:21
03 июл 2025, 13:21
c4a3352
Код
Авторство
О чём код?
from langchain_ollama import OllamaLLM from langchain.agents import AgentExecutor, create_structured_chat_agent, tool, initialize_agent, AgentType from langchain_core.prompts import ChatPromptTemplate llm = OllamaLLM(model="llama3",base_url="http://localhost:11434") @tool def get_square(n: int) -> str: """Вернуть квадрат целого числа.""" n = int(n) return f"Квадрат числа {n} = {n*n}" @tool def reverse_text(text: str) -> str: """Развернуть строку задом-наперёд.""" return text[::-1] tools = [get_square, reverse_text] agent = initialize_agent( tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True, handle_parsing_errors=True ) print(agent.run("Разверни 'Оллама' и найди квадрат 7"))