AlessandroMasala commited on
Commit
b79fe38
·
verified ·
1 Parent(s): 4008369

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +59 -0
agent.py CHANGED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # --- IMPORTS --- #
2
+ import os
3
+ import re
4
+ import ast
5
+ import requests
6
+ import tempfile
7
+ from pathlib import Path
8
+ from enum import Enum
9
+ from typing import Optional, Dict, List, Any, TypedDict
10
+
11
+ from dotenv import load_dotenv
12
+ from langgraph.graph import StateGraph, END
13
+ from langchain.tools import Tool as LangTool
14
+ from langchain_core.runnables import RunnableLambda
15
+ from langchain.tools import StructuredTool
16
+
17
+ # --- Integrazione Zephyr LLM ---
18
+ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
19
+ from langchain_community.llms import HuggingFacePipeline
20
+ import torch
21
+
22
+
23
+
24
+
25
+
26
+
27
+
28
+
29
+
30
+
31
+
32
+
33
+
34
+
35
+
36
+
37
+
38
+
39
+
40
+
41
+
42
+
43
+
44
+
45
+
46
+
47
+
48
+
49
+
50
+
51
+
52
+
53
+
54
+
55
+
56
+
57
+
58
+
59
+