Spaces:
Running
Running
Update agents.py
Browse files
agents.py
CHANGED
|
@@ -5,7 +5,7 @@ import traceback
|
|
| 5 |
from dotenv import load_dotenv
|
| 6 |
|
| 7 |
from langfuse import Langfuse
|
| 8 |
-
from
|
| 9 |
from langfuse.callback import CallbackHandler
|
| 10 |
from langchain_core.prompts import PromptTemplate
|
| 11 |
|
|
@@ -25,19 +25,19 @@ langfuse_post_generation_handler = CallbackHandler(
|
|
| 25 |
)
|
| 26 |
langfuse = Langfuse()
|
| 27 |
|
| 28 |
-
analysis_llm =
|
| 29 |
-
model="
|
| 30 |
temperature=0.8,
|
| 31 |
timeout=None,
|
| 32 |
max_retries=2,
|
| 33 |
-
api_key=os.getenv("
|
| 34 |
)
|
| 35 |
-
post_content_llm =
|
| 36 |
-
model="
|
| 37 |
temperature=0.8,
|
| 38 |
timeout=None,
|
| 39 |
max_retries=2,
|
| 40 |
-
api_key=os.getenv("
|
| 41 |
)
|
| 42 |
|
| 43 |
|
|
@@ -110,4 +110,4 @@ def get_text_post_content(details, reference):
|
|
| 110 |
except Exception as e:
|
| 111 |
print(e)
|
| 112 |
traceback.print_exc()
|
| 113 |
-
return "", False
|
|
|
|
| 5 |
from dotenv import load_dotenv
|
| 6 |
|
| 7 |
from langfuse import Langfuse
|
| 8 |
+
from langchain_google_genai import ChatGoogleGenerativeAI
|
| 9 |
from langfuse.callback import CallbackHandler
|
| 10 |
from langchain_core.prompts import PromptTemplate
|
| 11 |
|
|
|
|
| 25 |
)
|
| 26 |
langfuse = Langfuse()
|
| 27 |
|
| 28 |
+
analysis_llm = ChatGoogleGenerativeAI(
|
| 29 |
+
model="gemini-2.0-flash",
|
| 30 |
temperature=0.8,
|
| 31 |
timeout=None,
|
| 32 |
max_retries=2,
|
| 33 |
+
api_key=os.getenv("GOOGLE_API_KEY"),
|
| 34 |
)
|
| 35 |
+
post_content_llm = ChatGoogleGenerativeAI(
|
| 36 |
+
model="gemini-2.0-flash",
|
| 37 |
temperature=0.8,
|
| 38 |
timeout=None,
|
| 39 |
max_retries=2,
|
| 40 |
+
api_key=os.getenv("GOOGLE_POST_GENERATION_API_KEY"),
|
| 41 |
)
|
| 42 |
|
| 43 |
|
|
|
|
| 110 |
except Exception as e:
|
| 111 |
print(e)
|
| 112 |
traceback.print_exc()
|
| 113 |
+
return "", False
|