dipan004 sahil-1-garg commited on
Commit
499af4f
·
verified ·
1 Parent(s): 042bad4

Update agents/simple_tools.py (#3)

Browse files

- Update agents/simple_tools.py (1e60d058963ee33e66e62ad3220ba452420000c9)


Co-authored-by: sahilgarg <sahil-1-garg@users.noreply.huggingface.co>

Files changed (1) hide show
  1. agents/simple_tools.py +4 -0
agents/simple_tools.py CHANGED
@@ -389,6 +389,10 @@ def generate_llm_notes(file_path: str, note_numbers: str = "", **kwargs) -> Dict
389
  env = os.environ.copy()
390
  if os.getenv("OPENROUTER_API_KEY"):
391
  env["OPENROUTER_API_KEY"] = os.getenv("OPENROUTER_API_KEY")
 
 
 
 
392
  cwd = os.getcwd()
393
 
394
  # Step 1: Run LLM notes data processor
 
389
  env = os.environ.copy()
390
  if os.getenv("OPENROUTER_API_KEY"):
391
  env["OPENROUTER_API_KEY"] = os.getenv("OPENROUTER_API_KEY")
392
+ # Override with user-provided API key if available
393
+ user_api_key = kwargs.get("user_api_key")
394
+ if user_api_key:
395
+ env["OPENROUTER_API_KEY"] = user_api_key
396
  cwd = os.getcwd()
397
 
398
  # Step 1: Run LLM notes data processor