bobobert4 commited on
Commit ·
1fd978d
1
Parent(s): 8a95271
fix agent file prompt
Browse files
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.
|
| 39 |
-
max_retries=
|
| 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
|
| 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}"""
|
|
|