bobobert4 commited on
Commit
1fd978d
·
1 Parent(s): 8a95271

fix agent file prompt

Browse files
Files changed (1) hide show
  1. agent.py +6 -7
agent.py CHANGED
@@ -25,8 +25,8 @@ from tools import basic_tools
25
  # chat = ChatHuggingFace(llm=llm, verbose=True)
26
 
27
  # Google's chat interface
28
- RPM = 9
29
- TPM = 250000
30
  limiter = InMemoryRateLimiter(
31
  requests_per_second=(RPM / 60),
32
  check_every_n_seconds=(RPM / 70),
@@ -34,9 +34,9 @@ limiter = InMemoryRateLimiter(
34
  )
35
  chat = ChatGoogleGenerativeAI(
36
  # model="gemini-2.0-flash-lite",
37
- model="gemini-2.5-flash-preview-04-17",
38
- temperature=0.3,
39
- max_retries=5,
40
  verbose=True,
41
  rate_limiter=limiter,
42
  )
@@ -88,5 +88,4 @@ def get_system_prompt(prompt_file: Path = None):
88
  return SystemMessage(content=system_prompt)
89
 
90
  def insert_file_into_query(query: str, file_name: str = ""):
91
- return f"""{query} - Adjacent files in path > {file_name}
92
- """
 
25
  # chat = ChatHuggingFace(llm=llm, verbose=True)
26
 
27
  # Google's chat interface
28
+ RPM = os.environ.get("AGENT_MODEL_RPM", 9)
29
+ TPM = os.environ.get("AGENT_MODEL_TPM", 250000)
30
  limiter = InMemoryRateLimiter(
31
  requests_per_second=(RPM / 60),
32
  check_every_n_seconds=(RPM / 70),
 
34
  )
35
  chat = ChatGoogleGenerativeAI(
36
  # model="gemini-2.0-flash-lite",
37
+ model=os.environ.get("AGENT_MODEL","gemini-2.5-flash-preview-04-17"),
38
+ temperature=os.environ.get("AGENT_MODEL_TEMP", 0.25),
39
+ max_retries=os.environ.get("AGENT_MODEL_RETRIES", 2),
40
  verbose=True,
41
  rate_limiter=limiter,
42
  )
 
88
  return SystemMessage(content=system_prompt)
89
 
90
  def insert_file_into_query(query: str, file_name: str = ""):
91
+ return f"""{query} - Adjacent files path > {file_name}"""