ДОбился запуска

This commit is contained in:
2026-04-01 00:53:21 +05:00
parent 0af4730b3c
commit 71891765fb
4 changed files with 80 additions and 75 deletions

View File

@@ -1,15 +1,13 @@
from typing import TypedDict, List, Optional, Annotated
import operator
from langchain_community.llms import Ollama
from langchain_community.embeddings import OllamaEmbeddings
from langchain_community.vectorstores import Qdrant
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain_community.document_loaders import DirectoryLoader, PyPDFLoader
from langgraph.graph import StateGraph, END
from langgraph.graph.message import add_messages
from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
from langchain_core.messages import AIMessage, SystemMessage
from langchain.tools import tool
from langchain.agents import AgentExecutor, create_react_agent
# from langchain.agents import AgentExecutor, create_react_agent
import numpy as np
# Определяем состояние агента
@@ -22,16 +20,11 @@ class AgentState(TypedDict):
# Инициализация LLM через Ollama
llm = Ollama(
model="llama3.2:3b",
model="qwen2.5-coder-16k:14b",
temperature=0.1, # Низкая температура для консистентности
num_predict=1024, # Максимальная длина ответа
)
# Инициализация эмбеддингов
embeddings = OllamaEmbeddings(
model="nomic-embed-text", # Хорошие локальные эмбеддинги
)
# Создаем инструменты для агента
@tool
def search_knowledge_base(query: str) -> str:
@@ -127,6 +120,3 @@ def create_agent_graph():
workflow.add_edge("generate", END)
return workflow.compile()
# Создаем и компилируем граф
agent_graph = create_agent_graph()