update deps and tests

valory-xyz · Jun 8, 2024 · 8042ba6 · 8042ba6
1 parent 625fab0
commit 8042ba6
Show file tree

Hide file tree

Showing 8 changed files with 503 additions and 454 deletions.
diff --git a/.github/workflows/common_checks.yaml b/.github/workflows/common_checks.yaml
@@ -191,6 +191,7 @@ jobs:
           NEWS_API_KEY: ${{ secrets.NEWS_API_KEY }}
           OPENROUTER_API_KEY: ${{ secrets.OPEN_ROUTER_API_KEY }}
           GNOSIS_RPC_URL: ${{ secrets.GNOSIS_RPC_URL }}
+          GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }}
         run: |
           printenv
           tox -e check-tools
diff --git a/packages/packages.json b/packages/packages.json
@@ -21,13 +21,13 @@
         "custom/napthaai/prediction_url_cot_claude/0.1.0": "bafybeicbjywni5hx5ssoiv6tnnjbqzsck6cmtsdpr6m562z6afogz5eh44",
         "custom/napthaai/prediction_request_reasoning_claude/0.1.0": "bafybeihtx2cejxoy42jwk2i5m4evfzz537aic5njuawxnzdzwlo63kdduq",
         "custom/napthaai/prediction_request_rag_claude/0.1.0": "bafybeickr32t7nmapuoymjyo3cf5rr2v2zapksxcivuqsgjr2gn6zo6y7y",
-        "custom/napthaai/prediction_request_rag_cohere/0.1.0": "bafybeicrpk5oktsmr4bnk5flpkfqfwbcttq6n2oo742334d2iaixu6w5qa",
+        "custom/napthaai/prediction_request_rag_cohere/0.1.0": "bafybeihh26kwgqkzojd4zastto5mgod7o7esll4f4icurvwumvwuzmosrm",
         "custom/valory/prediction_request_lite/0.1.0": "bafybeidlrn4siraksldhiq3vbhhil4lbzuaqzooppkfhfql5rzii6cnziq",
         "custom/nickcom007/prediction_request_sme_lite/0.1.0": "bafybeidhuikn322h7oxgl5n2t5lvcl2dmv7cbmnes5tomabuucg6a42zkm",
         "custom/napthaai/prediction_request_reasoning_lite/0.1.0": "bafybeigxs5tq4w7ouamwlvv7vjw3z3jeercynsuka4mcpveshopvej4cyu",
-        "custom/napthaai/prediction_request_rag_cohere/0.1.0": "bafybeihh26kwgqkzojd4zastto5mgod7o7esll4f4icurvwumvwuzmosrm",
         "custom/valory/prediction_langchain/0.1.0": "bafybeihhii7veepp6ovkmqjnkp6euhkwm52obabgdltdj34ikisfd7yvqi",
         "custom/victorpolisetty/gemini_request/0.1.0": "bafybeig5x6b5jtanet2q5sk7er7fdzpippbvh4q5p7uxmxpriq66omjnaq",
+        "custom/gnosis/omen_tools/0.1.0": "bafybeifxrawgu6m3dgsxvj7jrhxzr5gwi3zjk2m4gltkr5w3hxjjbla6nu",
         "protocol/valory/acn_data_share/0.1.0": "bafybeih5ydonnvrwvy2ygfqgfabkr47s4yw3uqxztmwyfprulwfsoe7ipq",
         "protocol/valory/websocket_client/0.1.0": "bafybeifjk254sy65rna2k32kynzenutujwqndap2r222afvr3zezi27mx4",
         "contract/valory/agent_mech/0.1.0": "bafybeiah6b5epo2hlvzg5rr2cydgpp2waausoyrpnoarf7oa7bw33rex34",
@@ -40,8 +40,8 @@
         "skill/valory/task_execution/0.1.0": "bafybeic27brlpaamu3gfmdribywjvjyzxz2evqneh3e2kb3karubllz6xi",
         "skill/valory/websocket_client/0.1.0": "bafybeif7rrvsu6z4evqkhblxj3u6wwv2eqou576hgkyoehxuj7cntw7o2m",
         "skill/valory/subscription_abci/0.1.0": "bafybeihtortmv4fqua5wrnshpnvqsbpaf52frwynrmpuv2uw5j7wkauhze",
-        "agent/valory/mech/0.1.0": "bafybeid2hlmwtoze3xhhqqayisdg6xzxzgf42zw7hccdaszur7qfgp5vu4",
-        "service/valory/mech/0.1.0": "bafybeibr425raoxalifrnspyph4r2u2ilijlvdm3yexxuzsmwzyx7vlica"
+        "agent/valory/mech/0.1.0": "bafybeih2oex4yt4mmiyarp2ivkqqfscoavyb7metchciiqmqdmv7lhyutq",
+        "service/valory/mech/0.1.0": "bafybeic4lql2xbqfgjyyiacsthfkkx5zea2khackffhoh36hhajhjyhgdi"
     },
     "third_party": {
         "protocol/valory/default/1.0.0": "bafybeifqcqy5hfbnd7fjv4mqdjrtujh2vx3p2xhe33y67zoxa6ph7wdpaq",

diff --git a/packages/valory/agents/mech/aea-config.yaml b/packages/valory/agents/mech/aea-config.yaml
@@ -138,6 +138,8 @@ dependencies:
   prediction-market-agent-tooling:
     git: https://github.com/gnosis/prediction-market-agent-tooling
     ref: gabriel/olas-deps
+  google-generativeai:
+    version: ==0.6.0
 default_connection: null
 ---
 public_id: valory/websocket_client:0.1.0:bafybeiexove4oqyhoae5xmk2hilskthosov5imdp65olpgj3cfrepbouyy

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -75,6 +75,7 @@ lxml = {extras = ["html-clean"], version = "^5.1.0"}
 langgraph = "==0.0.55"
 langchain-community = "==0.2.1"
 prediction-market-agent-tooling = {git = "https://github.com/gnosis/prediction-market-agent-tooling.git", rev = "gabriel/olas-deps"}
+google-generativeai = "==0.6.0"
 
 [tool.poetry.group.dev.dependencies]
 tomte = {version = "==0.2.15", extras = ["cli", "tests"]}
diff --git a/tests/constants.py b/tests/constants.py
@@ -33,3 +33,4 @@
 NEWS_API_KEY = os.getenv("NEWS_API_KEY")
 OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
 GNOSIS_RPC_URL = os.getenv("GNOSIS_RPC_URL")
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
diff --git a/tests/test_tools.py b/tests/test_tools.py
@@ -41,6 +41,7 @@
     NEWS_API_KEY,
     OPENROUTER_API_KEY,
     GNOSIS_RPC_URL,
+    GEMINI_API_KEY
 )
 
 
@@ -58,6 +59,7 @@ class BaseToolTest:
             "newsapi": [NEWS_API_KEY],
             "openrouter": [OPENROUTER_API_KEY],
             "gnosis_rpc_url": [GNOSIS_RPC_URL],
+            "gemini": [GEMINI_API_KEY],
         }
     )
     models: List = [None]

diff --git a/tox.ini b/tox.ini
@@ -70,6 +70,7 @@ deps =
     moviepy==1.0.3
     replicate==0.15.7
     prediction-market-agent-tooling: git+https://github.com/gnosis/prediction-market-agent-tooling.git@gabriel/olas-deps
+    google-generativeai==0.6.0
 
 [extra-deps]
 deps =