Skip to content

Commit

Permalink
#42 Adjust AI amount guidelines
Browse files Browse the repository at this point in the history
  • Loading branch information
Ivan Seredkin committed Jan 7, 2025
1 parent 509aade commit ed8c002
Showing 2 changed files with 8 additions and 7 deletions.
5 changes: 3 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -84,6 +84,7 @@ Analyze the provided portfolio and watchlist data to recommend:
- Sell Amounts Guidelines: Minimum 1.0 USD, Maximum 300.0 USD
- Buy Amounts Guidelines: Minimum 1.0 USD, Maximum 300.0 USD
- Stocks under PDT Limit: NVDA, MSFT, SNAP, NWSA, ...
- Trade Exceptions (exclude from trading in any decisions): AAPL, TSLA, AMZN, ...
**Portfolio Overview:**
```json
@@ -181,8 +182,8 @@ You are an investment advisor tasked with reviewing and adjusting prior trading
**Constraints:**
- Maintain a portfolio size of fewer than 10 stocks.
- Total Buying Power: 2.95 USD initially.
- Sell Amounts Guidelines: Minimum 1.0 USD, Maximum 300.0 USD
- Buy Amounts Guidelines: Minimum 1.0 USD, Maximum 300.0 USD
- Sell Amounts Guidelines: Minimum amount 1.0 USD, Maximum amount 300.0 USD
- Buy Amounts Guidelines: Minimum amount 1.0 USD, Maximum amount 300.0 USD
- Stocks under PDT Limit: NVDA, MSFT, SNAP, NWSA, ...
**Trading Results:**
10 changes: 5 additions & 5 deletions main.py
Original file line number Diff line number Diff line change
@@ -27,7 +27,7 @@ def parse_ai_response(ai_response):
try:
ai_content = re.sub(r'```json|```', '', ai_response.choices[0].message.content.strip())
decisions = json.loads(ai_content)
except json.JSONDecodeError as e:
except json.JSONDecodeError:
raise Exception("Invalid JSON response from OpenAI: " + ai_response.choices[0].message.content.strip())
return decisions

@@ -36,16 +36,16 @@ def parse_ai_response(ai_response):
def get_ai_amount_guidelines():
sell_guidelines = []
if MIN_SELLING_AMOUNT_USD is not False:
sell_guidelines.append(f"Minimum {MIN_SELLING_AMOUNT_USD} USD")
sell_guidelines.append(f"Minimum amount {MIN_SELLING_AMOUNT_USD} USD")
if MAX_SELLING_AMOUNT_USD is not False:
sell_guidelines.append(f"Maximum {MAX_SELLING_AMOUNT_USD} USD")
sell_guidelines.append(f"Maximum amount {MAX_SELLING_AMOUNT_USD} USD")
sell_guidelines = ", ".join(sell_guidelines) if sell_guidelines else None

buy_guidelines = []
if MIN_BUYING_AMOUNT_USD is not False:
buy_guidelines.append(f"Minimum {MIN_BUYING_AMOUNT_USD} USD")
buy_guidelines.append(f"Minimum amount {MIN_BUYING_AMOUNT_USD} USD")
if MAX_BUYING_AMOUNT_USD is not False:
buy_guidelines.append(f"Maximum {MAX_BUYING_AMOUNT_USD} USD")
buy_guidelines.append(f"Maximum amount {MAX_BUYING_AMOUNT_USD} USD")
buy_guidelines = ", ".join(buy_guidelines) if buy_guidelines else None

return sell_guidelines, buy_guidelines

0 comments on commit ed8c002

Please sign in to comment.