AlessandroMasala's picture
Update agent.py
cbd635f verified
raw
history blame
620 Bytes
# --- IMPORTS --- #
import os
import re
import ast
import requests
import tempfile
from pathlib import Path
from enum import Enum
from typing import Optional, Dict, List, Any, TypedDict
from dotenv import load_dotenv
from langgraph.graph import StateGraph, END
from langchain.tools import Tool as LangTool
from langchain_core.runnables import RunnableLambda
from langchain.tools import StructuredTool
# --- Integrazione Zephyr LLM ---
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
from langchain_community.llms import HuggingFacePipeline
import torch
#