mirror of
https://github.com/Xe138/AI-Trader.git
synced 2026-04-02 01:27:24 -04:00
Compare commits
2 Commits
v0.3.0-alp
...
v0.3.0-alp
| Author | SHA1 | Date | |
|---|---|---|---|
| bf12e981fe | |||
| a16bac5d08 |
@@ -45,11 +45,14 @@ class ContextInjector:
|
||||
"""
|
||||
# Inject signature and today_date for trade tools
|
||||
if request.name in ["buy", "sell"]:
|
||||
# Add signature and today_date to arguments if not present
|
||||
if "signature" not in request.arguments:
|
||||
request.arguments["signature"] = self.signature
|
||||
if "today_date" not in request.arguments:
|
||||
request.arguments["today_date"] = self.today_date
|
||||
# Add signature and today_date to args if not present
|
||||
if "signature" not in request.args:
|
||||
request.args["signature"] = self.signature
|
||||
if "today_date" not in request.args:
|
||||
request.args["today_date"] = self.today_date
|
||||
|
||||
# Debug logging
|
||||
print(f"[ContextInjector] Tool: {request.name}, Args after injection: {request.args}")
|
||||
|
||||
# Call the actual tool handler
|
||||
return await handler(request)
|
||||
|
||||
@@ -44,8 +44,10 @@ def buy(symbol: str, amount: int, signature: str = None, today_date: str = None)
|
||||
"""
|
||||
# Step 1: Get environment variables and basic information
|
||||
# Get signature (model name) from parameter or fallback to config/env
|
||||
print(f"[buy] Received signature parameter: {signature}")
|
||||
if signature is None:
|
||||
signature = get_config_value("SIGNATURE")
|
||||
print(f"[buy] Signature from config: {signature}")
|
||||
if signature is None:
|
||||
raise ValueError("SIGNATURE not provided and environment variable is not set")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user