memory.py does not seem to handle any other llm than openai

[axlesor]

I used Gemini as llm by running cli.main and always got:
NotFoundError: Error code: 404
[NOTE] During task with name 'Bull Researcher'

Just checking code I believe the issue is in file: agents/utils/memory.py assumes backend llm is openapi and Gemini
does not understand "text-embedding-3-small"
As much as I follow Gemini uses self.embedding = "text-embedding-004:embedContent"
but on a different endpoint, hence changes are not that simple and Gemini suggested:
Summary of Changes Required
Change Endpoint: Switch from /v1/embeddings to /v1beta/models/text-embedding-004:embedContent.
Change Model: Use text-embedding-004 instead of text-embedding-3-small.
Update Payload: Change {"input": "..."} to {"content": {"parts": [{"text": "..."}]}}.

I guess I can get openai APi key to pass this issue now.

[axlesor]

fyi: found another open issue called: when use google llm, encounter 404 error #95
which is the same error as above. And following recommendations from that thread
here is simple patch that fixed the issue running with Gemini (no need for openai api key):
'''
$ git diff
diff --git a/cli/utils.py b/cli/utils.py
index 7b9682a..4cca192 100644
--- a/cli/utils.py
+++ b/cli/utils.py
@@ -245,7 +245,7 @@ def select_llm_provider() -> tuple[str, str]:
BASE_URLS = [
("OpenAI", "https://api.openai.com/v1"),
("Anthropic", "https://api.anthropic.com/"),
- ("Google", "https://generativelanguage.googleapis.com/v1"),
+ ("Google", "https://generativelanguage.googleapis.com/v1beta"),
("Openrouter", "https://openrouter.ai/api/v1"),
("Ollama", "http://localhost:11434/v1"),
]
diff --git a/tradingagents/agents/utils/memory.py b/tradingagents/agents/utils/memory.py
index 69b8ab8..5f9bb9f 100644
--- a/tradingagents/agents/utils/memory.py
+++ b/tradingagents/agents/utils/memory.py
@@ -7,6 +7,8 @@ class FinancialSituationMemory:
def init(self, name, config):
if config["backend_url"] == "http://localhost:11434/v1":
self.embedding = "nomic-embed-text"
+ elif config["backend_url"] == "https://generativelanguage.googleapis.com/v1beta":
+ self.embedding = "gemini-embedding-exp-03-07"
else:
self.embedding = "text-embedding-3-small"
self.client = OpenAI(base_url=config["backend_url"])
'''