RAG_Agent / globals.py
ArseniyPerchik's picture
more
e758f09
from typing import TypedDict, Annotated
import os
import random
from lmnr import Laminar
from dotenv import load_dotenv
load_dotenv()
import datasets
from huggingface_hub import list_models
from langchain.docstore.document import Document
from langchain_community.retrievers import BM25Retriever
from langchain_community.tools import DuckDuckGoSearchRun
from langchain.tools import Tool
from langchain_ollama import ChatOllama
from langchain_huggingface import HuggingFaceEndpoint,ChatHuggingFace
from langchain_core.messages import AnyMessage, HumanMessage, AIMessage
from langgraph.graph import StateGraph, START, END
from langgraph.graph.message import add_messages
from langgraph.prebuilt import tools_condition, ToolNode
# GLOBALS
HF_TOKEN = os.getenv('HF_TOKEN')
PHOENIX_API_KEY = os.getenv('PHOENIX_API_KEY')
LANGFUSE_PUBLIC_KEY = os.getenv('LANGFUSE_PUBLIC_KEY')
LANGFUSE_SECRET_KEY= os.getenv('LANGFUSE_SECRET_KEY')
LANGFUSE_HOST= os.getenv('LANGFUSE_HOST')
LAMINAR_API_KEY= os.getenv('LAMINAR_API_KEY')