pritmanvar-bacancy commited on
Commit
0e5c361
·
verified ·
1 Parent(s): 179e2f6

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +8 -8
agents.py CHANGED
@@ -5,7 +5,7 @@ import traceback
5
  from dotenv import load_dotenv
6
 
7
  from langfuse import Langfuse
8
- from langchain_groq import ChatGroq
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 = ChatGroq(
29
- model="llama-3.1-8b-instant",
30
  temperature=0.8,
31
  timeout=None,
32
  max_retries=2,
33
- api_key=os.getenv("GROQ_ANALYSIS_API_KEY"),
34
  )
35
- post_content_llm = ChatGroq(
36
- model="qwen/qwen3-32b",
37
  temperature=0.8,
38
  timeout=None,
39
  max_retries=2,
40
- api_key=os.getenv("THREADS_POST_GENERATION_API_KEY"),
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