increased httpx timeout to 60 secs

.chainlit/config.toml

@@ -10,7 +10,7 @@ user_env = []
 session_timeout = 3600
 
 # Enable third parties caching (e.g LangChain cache)
-cache = false
+cache = true
 
 # Authorized origins
 allow_origins = ["*"]
@@ -33,7 +33,7 @@ auto_tag_thread = true
 
 # Authorize users to spontaneously upload files with messages
 [features.spontaneous_file_upload]
-    enabled = true
+    enabled = false
     accept = ["*/*"]
     max_files = 20
     max_size_mb = 500

chatxbt-assistant.py

@@ -1,20 +1,14 @@
-import os, json
+import os
 from typing import Optional, Dict
 from dotenv import load_dotenv
-
-load_dotenv()
-
 import chainlit as cl
 from phi.assistant import Assistant
 from phi.llm.openai import OpenAIChat
-from phi.llm.ollama import Ollama
-from phi.llm.groq import Groq
 from phi.tools.duckduckgo import DuckDuckGo
 from phi.tools.yfinance import YFinanceTools
 from src.databases.postgres import sqlalchemy_engine
 from src.knowledge_bases.combined import knowledge_base
 from src.tools.crypto_swap_toolkit import CryptoSwapTools
-from src.tools.crypto_bridge_toolkit import CrossChainSwapTools
 from src.tools.crypto_data_toolkit import CryptoDataTools
 from portkey_ai import PORTKEY_GATEWAY_URL, createHeaders
 from src.config.portkey_config import generate_portkey_config
@@ -23,33 +17,28 @@ from phi.storage.assistant.postgres import PgAssistantStorage
 from src.tools.crypto_evm_wallet_toolkit import CryptoEVMWalletTools
 from src.tools.user_confirmation_pin_toolkit import UserConfirmationPinToolkit
 
+# Load environment variables
+load_dotenv()
 
+# Initialize storage
 storage = PgAssistantStorage(
-    # stores runs in the ai.assistant_runs table
     table_name="assistant_runs",
     db_engine=sqlalchemy_engine
 )
 
+# OAuth callback
 @cl.oauth_callback
-def oauth_callback(
-  provider_id: str,
-  token: str,
-  raw_user_data: Dict[str, str],
-  default_user: cl.User,
-) -> Optional[cl.User]:
-    # step-todo: will handle auth from v2 api
+def oauth_callback(provider_id: str, token: str, raw_user_data: Dict[str, str], default_user: cl.User) -> Optional[cl.User]:
     return cl.User(
         identifier=raw_user_data.get('email'),
         email=raw_user_data.get('email'),
         name=raw_user_data.get('name'),
-        metadata={**raw_user_data, **({'image':default_user.metadata.get('image')})}
+        metadata={**raw_user_data, **({'image': default_user.metadata.get('image')})}
     )
 
-
+# Password authentication callback
 @cl.password_auth_callback
-def auth_callback(username: str, password: str):
-    # Fetch the user matching username from your database
-    # and compare the hashed password with the value stored in the database
+def auth_callback(username: str, password: str) -> Optional[cl.User]:
     if (username, password) == ("admin", "admin"):
         user = {
             "identifier": "admin", "email": username, "metadata": {"role": "admin", "provider": "credentials"}
@@ -58,10 +47,9 @@ def auth_callback(username: str, password: str):
         return cl.User(
             identifier="admin", email=username, metadata={"role": "admin", "provider": "credentials"}
         )
-    else:
-        return None
-
+    return None
 
+# Set starters
 @cl.set_starters
 async def set_starters():
     return [
@@ -70,41 +58,46 @@ async def set_starters():
             message="create a crypto wallet for me",
             icon="/public/wallet-svgrepo-com.svg",
         ),
-
         cl.Starter(
             label="Latest News on defi, crypto and solana",
-            message="What news are currentlly trending on defi.",
+            message="What news are currently trending on defi.",
             icon="/public/news-svgrepo-com.svg",
-            ),
+        ),
         cl.Starter(
             label="Get price of BTC, ETH and PEPE",
-            message="Get me the Price of BTC, ETH and PEPE",
+            message="Get me the price of BTC, ETH and PEPE",
             icon="/public/coins-electronics-svgrepo-com.svg",
-            ),
+        ),
         cl.Starter(
             label="Get trending stocks",
             message="Get latest stock",
             icon="/public/stockchart-svgrepo-com.svg",
-            )
-        ]
+        )
+    ]
 
+# On chat start
 @cl.on_chat_start
 async def start():
-    is_dev_mode = True if os.getenv("DEV_MODE") else False
-
-    portkey_local_gateway = True if os.getenv("PORTKEY_LOCAL_GATEWAY_URL") else False
+    is_dev_mode = bool(os.getenv("DEV_MODE"))
+    portkey_local_gateway = bool(os.getenv("PORTKEY_LOCAL_GATEWAY_URL"))
     portkey_config = generate_portkey_config(local=portkey_local_gateway)
 
     # Initialize the assistant
     cxbt_assistant = Assistant(
-        # llm=OpenAIChat(),
-        # llm=Groq(model="llama3-70b-8192"),
-        llm=Ollama(model="llama3"),
+        introduction="Hi, I'm ChatXBT, your AI assistant for Web3 and DeFi. I can help you with your queries related to Web3 and DeFi.",
+        llm=OpenAIChat(
+            api_key=os.getenv("PORTKEY_API_KEY"),
+            # base_url=os.getenv("PORTKEY_LOCAL_GATEWAY_URL") or PORTKEY_GATEWAY_URL,
+            base_url=PORTKEY_GATEWAY_URL,
+            default_headers=createHeaders(
+                api_key=os.getenv("PORTKEY_API_KEY") or None,
+                config=portkey_config
+            ),
+        ),
         tools=[
             DuckDuckGo(),
             CryptoDataTools(),
             CryptoSwapTools(),
-            CrossChainSwapTools(),
             UserProfileToolkit(),
             CryptoEVMWalletTools(),
             UserConfirmationPinToolkit(),
@@ -119,31 +112,34 @@ async def start():
         add_references_to_prompt=True,
         add_chat_history_to_prompt=True,
         prevent_hallucinations=True,
-        prevent_prompt_injection=True
+        prevent_prompt_injection=True,
+        add_datetime_to_instructions=True
     )
     cxbt_assistant.knowledge_base.load(recreate=False)
 
     # Set the assistant in the user session
     cl.user_session.set("agent", cxbt_assistant)
 
+# On message
 @cl.on_message
 async def main(message: cl.Message):
-
     msg = cl.Message(content="")
-    await msg.send()
 
     # Retrieve the assistant from the user session
     agent = cl.user_session.get("agent")
 
     # Process the user message using the assistant
-    for delta in agent.run(message.content, stream=True):
-        for part in delta:
-            if token := part or "":
-                # Send the response back to the user
-                await msg.stream_token(token)
+    try:
+        response_generator = agent.run(message=message.content, stream=True)
+        for delta in response_generator:
+            await msg.stream_token(str(delta))
+    except TypeError as e:
+        # Handle specific TypeError and log or print additional information for debugging
+        print(f"Error occurred: {e}")
+        await msg.stream_token(f"\n\n I encountetrd an error, please try again later.")
 
-    await msg.update()
+    await msg.send()
 
 # Run the Chainlit application
 if __name__ == "__main__":
-    cl.run()
+    cl.run_sync()

lite-llm-config.yaml

@@ -1,15 +0,0 @@
-model_list: 
-  - model_name: gpt-3.5-turbo # user-facing model alias
-    litellm_params: # all params accepted by litellm.completion() - https://docs.litellm.ai/docs/completion/input
-      model: azure/<your-deployment-name>
-      api_base: <your-azure-api-endpoint>
-      api_key: <your-azure-api-key>
-  - model_name: gpt-3.5-turbo
-    litellm_params:
-      model: azure/gpt-turbo-small-ca
-      api_base: https://my-endpoint-canada-berri992.openai.azure.com/
-      api_key: <your-azure-api-key>
-  - model_name: vllm-model
-    litellm_params:
-      model: openai/<your-model-name>
-      api_base: <your-api-base> # e.g. http://0.0.0.0:3000

requirements.bk.txt

@@ -1,52 +1,134 @@
+aiobotocore==2.13.1
+aiofiles==23.2.1
 aiohttp==3.9.5
+aioitertools==0.11.0
 aiosignal==1.3.1
 annotated-types==0.7.0
-anyio==4.4.0
+anthropic==0.31.1
+anyio==3.7.1
+appdirs==1.4.4
 appnope==0.1.4
+APScheduler==3.10.4
 asgiref==3.8.1
 asttokens==2.4.1
 async-timeout==4.0.3
+asyncer==0.0.2
 attrs==23.2.0
 backcall==0.2.0
+backoff==2.2.1
+baize==0.20.8
 beautifulsoup4==4.12.3
+bidict==0.23.1
+bitarray==2.9.2
 bleach==6.1.0
-certifi==2024.6.2
+boto3==1.34.131
+botocore==1.34.131
+cached-property==1.5.2
+certifi==2024.7.4
+cffi==1.16.0
+chainlit==1.1.306
 charset-normalizer==3.3.2
+chevron==0.14.0
+ckzg==1.0.2
+click==8.1.7
 coingecko==0.13
+cryptography==42.0.8
+cytoolz==0.12.3
+dataclasses-json==0.5.14
 decorator==5.1.1
 defusedxml==0.7.1
 Deprecated==1.2.14
+distro==1.9.0
+dnspython==2.6.1
 docopt==0.6.2
-exceptiongroup==1.2.1
+duckduckgo_search==6.2.1
+email_validator==2.2.0
+eth-account==0.13.0
+eth-hash==0.7.0
+eth-keyfile==0.8.1
+eth-keys==0.5.1
+eth-rlp==2.1.0
+eth-typing==4.4.0
+eth-utils==4.1.1
+eth_abi==5.1.0
+exceptiongroup==1.2.2
 executing==2.0.1
-fastjsonschema==2.19.1
+fastapi==0.110.3
+fastapi-cli==0.0.4
+fastapi-sso==0.15.0
+fastjsonschema==2.20.0
+filelock==3.15.4
+filetype==1.2.0
+frozendict==2.4.4
 frozenlist==1.4.1
-googleapis-common-protos==1.63.1
+fs==2.4.16
+fs-s3fs==1.1.1
+fsspec==2024.6.1
+gitdb==4.0.11
+GitPython==3.1.43
+google_search_results==2.4.2
+googleapis-common-protos==1.63.2
+greenlet==3.0.3
+grpcio==1.64.1
+gunicorn==22.0.0
 h11==0.14.0
+hexbytes==1.2.1
+html5lib==1.1
 httpcore==1.0.5
+httptools==0.6.1
 httpx==0.27.0
+huggingface-hub==0.23.5
 idna==3.7
 importlib_metadata==7.1.0
 ipython==8.12.3
 jedi==0.19.1
 Jinja2==3.1.4
-jsonschema==4.22.0
+jiter==0.5.0
+jmespath==1.0.1
+jsonpatch==1.33
+jsonpointer==3.0.0
+jsonschema==4.23.0
 jsonschema-specifications==2023.12.1
 jupyter_client==8.6.2
 jupyter_core==5.7.2
 jupyterlab_pygments==0.3.0
-logfire==0.42.0
+langchain==0.2.8
+langchain-anthropic==0.1.20
+langchain-community==0.2.7
+langchain-core==0.2.20
+langchain-openai==0.1.16
+langchain-text-splitters==0.2.2
+langchainhub==0.1.20
+langgraph==0.1.8
+langsmith==0.1.88
+Lazify==0.4.0
+litellm==1.41.23
+literalai==0.0.607
+logfire==0.46.1
+lru-dict==1.3.0
+lxml==5.2.2
 markdown-it-py==3.0.0
 MarkupSafe==2.1.5
+marshmallow==3.21.3
 matplotlib-inline==0.1.7
 mdurl==0.1.2
 mistune==3.0.2
 multidict==6.0.5
+multitasking==0.0.11
+mypy==1.10.1
+mypy-extensions==1.0.0
 nbclient==0.10.0
 nbconvert==7.16.4
 nbformat==5.10.4
+nest-asyncio==1.6.0
+numexpr==2.10.1
+numpy==1.26.4
+oauthlib==3.2.2
+openai==1.35.14
 opentelemetry-api==1.25.0
+opentelemetry-exporter-otlp==1.25.0
 opentelemetry-exporter-otlp-proto-common==1.25.0
+opentelemetry-exporter-otlp-proto-grpc==1.25.0
 opentelemetry-exporter-otlp-proto-http==1.25.0
 opentelemetry-instrumentation==0.46b0
 opentelemetry-instrumentation-asgi==0.46b0
@@ -59,46 +141,99 @@ opentelemetry-proto==1.25.0
 opentelemetry-sdk==1.25.0
 opentelemetry-semantic-conventions==0.46b0
 opentelemetry-util-http==0.46b0
-packaging==24.1
+orjson==3.10.6
+packaging==23.2
+pandas==2.2.2
 pandocfilters==1.5.1
+parsimonious==0.10.0
 parso==0.8.4
+peewee==3.17.6
 pexpect==4.9.0
+pgvector==0.3.1
+phidata==2.4.25
 pickleshare==0.7.5
-pipdeptree==2.22.0
+pipdeptree==2.23.1
 pipreqs==0.5.0
 platformdirs==4.2.2
+portkey-ai==1.7.0
 prompt_toolkit==3.0.47
 protobuf==4.25.3
 psutil==5.9.8
+psycopg==3.2.1
+psycopg2==2.9.9
 ptyprocess==0.7.0
 pure-eval==0.2.2
-pydantic==2.7.4
-pydantic_core==2.18.4
+pycparser==2.22
+pycryptodome==3.20.0
+pydantic==2.8.2
+pydantic-settings==2.3.4
+pydantic_core==2.20.1
 Pygments==2.18.0
+PyJWT==2.8.0
+pypeln==0.4.9
+pyreqwest_impersonate==0.5.0
 python-dateutil==2.9.0.post0
 python-dotenv==1.0.1
-python-ulid==2.6.0
+python-engineio==4.9.1
+python-multipart==0.0.9
+python-socketio==5.11.3
+python-ulid==2.7.0
+pytz==2024.1
+pyunormalize==15.1.0
+PyYAML==6.0.1
 pyzmq==26.0.3
-redis==5.0.6
+redis==5.0.7
 referencing==0.35.1
+regex==2024.5.15
 requests==2.32.3
 rich==13.7.1
-rpds-py==0.18.1
+rlp==4.0.1
+rpc.py==0.6.0
+rpds-py==0.19.0
+rq==1.16.2
+s3fs==2024.6.1
+s3transfer==0.10.2
+setuptools==70.3.0
+shellingham==1.5.4
+simple-websocket==1.0.0
 six==1.16.0
+smmap==5.0.1
 sniffio==1.3.1
 soupsieve==2.5
+SQLAlchemy==2.0.31
 stack-data==0.6.3
+starlette==0.37.2
+stopit==1.1.2
+syncer==2.0.3
+tenacity==8.5.0
+tiktoken==0.7.0
 tinycss2==1.3.0
+tokenizers==0.19.1
 tomli==2.0.1
+toolz==0.12.1
 tornado==6.4.1
+tqdm==4.66.4
 traitlets==5.14.3
+typer==0.12.3
+types-requests==2.32.0.20240712
+typing-inspect==0.9.0
 typing_extensions==4.12.2
+tzdata==2024.1
+tzlocal==5.2
 ulid==1.1
 upstash-redis==1.1.0
-urllib3==2.2.1
+uptrace==1.24.0
+urllib3==2.2.2
+uvicorn==0.25.0
+uvloop==0.19.0
+watchfiles==0.20.0
 wcwidth==0.2.13
+web3==6.11.0
 webencodings==0.5.1
+websockets==12.0
 wrapt==1.16.0
+wsproto==1.2.0
 yarg==0.1.9
 yarl==1.9.4
+yfinance==0.2.40
 zipp==3.19.2

requirements.txt

@@ -1,10 +1,10 @@
-aiobotocore==2.13.0
+aiobotocore==2.13.1
 aiofiles==23.2.1
 aiohttp==3.9.5
 aioitertools==0.11.0
 aiosignal==1.3.1
 annotated-types==0.7.0
-anthropic==0.28.0
+anthropic==0.31.1
 anyio==3.7.1
 appdirs==1.4.4
 appnope==0.1.4
@@ -21,12 +21,12 @@ beautifulsoup4==4.12.3
 bidict==0.23.1
 bitarray==2.9.2
 bleach==6.1.0
-boto3==1.34.106
-botocore==1.34.106
+boto3==1.34.131
+botocore==1.34.131
 cached-property==1.5.2
-certifi==2024.6.2
+certifi==2024.7.4
 cffi==1.16.0
-chainlit==1.1.300
+chainlit==1.1.306
 charset-normalizer==3.3.2
 chevron==0.14.0
 ckzg==1.0.2
@@ -41,71 +41,71 @@ Deprecated==1.2.14
 distro==1.9.0
 dnspython==2.6.1
 docopt==0.6.2
-duckduckgo_search==6.1.6
+duckduckgo_search==6.2.1
 email_validator==2.2.0
-eth-account==0.11.2
+eth-account==0.13.0
 eth-hash==0.7.0
 eth-keyfile==0.8.1
 eth-keys==0.5.1
-eth-rlp==1.0.1
-eth-typing==4.3.1
+eth-rlp==2.1.0
+eth-typing==4.4.0
 eth-utils==4.1.1
 eth_abi==5.1.0
-exceptiongroup==1.2.1
+exceptiongroup==1.2.2
 executing==2.0.1
-fastapi==0.111.1
+fastapi==0.110.3
 fastapi-cli==0.0.4
-fastapi-sso==0.10.0
-fastjsonschema==2.19.1
-filelock==3.15.1
+fastapi-sso==0.15.0
+fastjsonschema==2.20.0
+filelock==3.15.4
 filetype==1.2.0
 frozendict==2.4.4
 frozenlist==1.4.1
 fs==2.4.16
 fs-s3fs==1.1.1
-fsspec==2024.6.0
+fsspec==2024.6.1
 gitdb==4.0.11
 GitPython==3.1.43
 google_search_results==2.4.2
-googleapis-common-protos==1.63.1
+googleapis-common-protos==1.63.2
 greenlet==3.0.3
 grpcio==1.64.1
 gunicorn==22.0.0
 h11==0.14.0
-hexbytes==0.3.1
+hexbytes==1.2.1
 html5lib==1.1
 httpcore==1.0.5
 httptools==0.6.1
 httpx==0.27.0
-huggingface-hub==0.23.3
+huggingface-hub==0.23.5
 idna==3.7
 importlib_metadata==7.1.0
 ipython==8.12.3
 jedi==0.19.1
 Jinja2==3.1.4
-jiter==0.4.2
+jiter==0.5.0
 jmespath==1.0.1
 jsonpatch==1.33
 jsonpointer==3.0.0
-jsonschema==4.22.0
+jsonschema==4.23.0
 jsonschema-specifications==2023.12.1
 jupyter_client==8.6.2
 jupyter_core==5.7.2
 jupyterlab_pygments==0.3.0
-langchain==0.2.4
-langchain-anthropic==0.1.15
-langchain-community==0.2.4
-langchain-core==0.2.6
-langchain-openai==0.1.8
-langchain-text-splitters==0.2.1
+langchain==0.2.8
+langchain-anthropic==0.1.20
+langchain-community==0.2.7
+langchain-core==0.2.20
+langchain-openai==0.1.16
+langchain-text-splitters==0.2.2
 langchainhub==0.1.20
-langgraph==0.0.68
-langsmith==0.1.77
+langgraph==0.1.8
+langsmith==0.1.88
 Lazify==0.4.0
-litellm==1.41.21
-literalai==0.0.604
-logfire==0.42.0
-lru-dict==1.2.0
+litellm==1.41.23
+literalai==0.0.607
+logfire==0.46.1
+lru-dict==1.3.0
 lxml==5.2.2
 markdown-it-py==3.0.0
 MarkupSafe==2.1.5
@@ -121,10 +121,10 @@ nbclient==0.10.0
 nbconvert==7.16.4
 nbformat==5.10.4
 nest-asyncio==1.6.0
-numexpr==2.10.0
+numexpr==2.10.1
 numpy==1.26.4
 oauthlib==3.2.2
-openai==1.34.0
+openai==1.35.14
 opentelemetry-api==1.25.0
 opentelemetry-exporter-otlp==1.25.0
 opentelemetry-exporter-otlp-proto-common==1.25.0
@@ -141,70 +141,71 @@ opentelemetry-proto==1.25.0
 opentelemetry-sdk==1.25.0
 opentelemetry-semantic-conventions==0.46b0
 opentelemetry-util-http==0.46b0
-orjson==3.10.5
+orjson==3.10.6
 packaging==23.2
 pandas==2.2.2
 pandocfilters==1.5.1
 parsimonious==0.10.0
 parso==0.8.4
-peewee==3.17.5
+peewee==3.17.6
 pexpect==4.9.0
-pgvector==0.2.5
-phidata==2.4.20
+pgvector==0.3.1
+phidata==2.4.25
 pickleshare==0.7.5
-pipdeptree==2.22.0
+pipdeptree==2.23.1
 pipreqs==0.5.0
 platformdirs==4.2.2
 portkey-ai==1.7.0
 prompt_toolkit==3.0.47
 protobuf==4.25.3
 psutil==5.9.8
-psycopg==3.1.19
+psycopg==3.2.1
 psycopg2==2.9.9
 ptyprocess==0.7.0
 pure-eval==0.2.2
 pycparser==2.22
 pycryptodome==3.20.0
-pydantic==2.7.4
-pydantic-settings==2.3.3
-pydantic_core==2.18.4
+pydantic==2.8.2
+pydantic-settings==2.3.4
+pydantic_core==2.20.1
 Pygments==2.18.0
 PyJWT==2.8.0
 pypeln==0.4.9
-pyreqwest_impersonate==0.4.7
+pyreqwest_impersonate==0.5.0
 python-dateutil==2.9.0.post0
 python-dotenv==1.0.1
 python-engineio==4.9.1
 python-multipart==0.0.9
-python-socketio==5.11.2
-python-ulid==2.6.0
+python-socketio==5.11.3
+python-ulid==2.7.0
 pytz==2024.1
 pyunormalize==15.1.0
 PyYAML==6.0.1
 pyzmq==26.0.3
-redis==5.0.6
+redis==5.0.7
 referencing==0.35.1
 regex==2024.5.15
 requests==2.32.3
 rich==13.7.1
 rlp==4.0.1
 rpc.py==0.6.0
-rpds-py==0.18.1
+rpds-py==0.19.0
 rq==1.16.2
-s3fs==2024.6.0
-s3transfer==0.10.1
+s3fs==2024.6.1
+s3transfer==0.10.2
+setuptools==70.3.0
 shellingham==1.5.4
 simple-websocket==1.0.0
 six==1.16.0
 smmap==5.0.1
 sniffio==1.3.1
 soupsieve==2.5
-SQLAlchemy==2.0.30
+SQLAlchemy==2.0.31
 stack-data==0.6.3
 starlette==0.37.2
 stopit==1.1.2
 syncer==2.0.3
-tenacity==8.3.0
+tenacity==8.5.0
 tiktoken==0.7.0
 tinycss2==1.3.0
 tokenizers==0.19.1
@@ -214,7 +215,7 @@ tornado==6.4.1
 tqdm==4.66.4
 traitlets==5.14.3
 typer==0.12.3
-types-requests==2.32.0.20240602
+types-requests==2.32.0.20240712
 typing-inspect==0.9.0
 typing_extensions==4.12.2
 tzdata==2024.1
@@ -222,12 +223,12 @@ tzlocal==5.2
 ulid==1.1
 upstash-redis==1.1.0
 uptrace==1.24.0
-urllib3==2.2.1
-uvicorn==0.22.0
+urllib3==2.2.2
+uvicorn==0.25.0
 uvloop==0.19.0
 watchfiles==0.20.0
 wcwidth==0.2.13
-web3==6.19.0
+web3==6.11.0
 webencodings==0.5.1
 websockets==12.0
 wrapt==1.16.0

src/config/portkey_config.py

@@ -13,7 +13,6 @@ def generate_portkey_config(local: bool = True):
             },
             "retry": {
                 "attempts": 3,
-                "on_status_codes": [429]
             },
             "targets": [
                 # {
@@ -30,8 +29,8 @@ def generate_portkey_config(local: bool = True):
                     "api_key": os.getenv("ANTHROPIC_API_KEY"),
                     "override_params": {
                         "model": "claude-3-5-sonnet-20240620",
-                        "max_tokens": 1024,
-                        "temperature": 0
+                        # "max_tokens": 1024,
+                        # "temperature": 0
                     }
                 },
                 {
@@ -39,8 +38,8 @@ def generate_portkey_config(local: bool = True):
                     "api_key": os.getenv("OPENAI_API_KEY"),
                     "override_params": {
                         "model": "gpt-4o",
-                        "max_tokens": 1024,
-                        "temperature": 0
+                        # "max_tokens": 1024,
+                        # "temperature": 0
                     }
                 },
                 {
@@ -48,8 +47,8 @@ def generate_portkey_config(local: bool = True):
                     "api_key": os.getenv("ANTHROPIC_API_KEY"),
                     "override_params": {
                         "model": "claude-3-opus-20240229",
-                        "max_tokens": 1024,
-                        "temperature": 0
+                        # "max_tokens": 1024,
+                        # "temperature": 0
                     }
                 }
             ]
@@ -65,16 +64,14 @@ def generate_portkey_config(local: bool = True):
             },
             "retry": {
                 "attempts": 3,
-                "on_status_codes": [429]
             },
             "targets": [
-                
                 {
                     "virtual_key": os.getenv("PORTKEY_OPENAI_VIRTUAL_KEY"),
                     "override_params": {
                         "model": "gpt-4o",
-                        "max_tokens": 1024,
-                        "temperature": 0
+                        # "max_tokens": 1024,
+                        # "temperature": 0
                     }
                 },
                 {
@@ -82,24 +79,24 @@ def generate_portkey_config(local: bool = True):
                     "api_key": os.getenv("POETRY_ANTHROPIC_API_KEY"),
                     "override_params": {
                         "model": "claude-3-5-sonnet-20240620",
-                        "max_tokens": 1024,
-                        "temperature": 0
+                        # "max_tokens": 1024,
+                        # "temperature": 0
                     }
                 },
                 {
                     "virtual_key": os.getenv("PORTKEY_ANTHROPIC_API_KEY"),
                     "override_params": {
                         "model": "claude-3-opus-20240229",
-                        "max_tokens": 1024,
-                        "temperature": 0
+                        # "max_tokens": 1024,
+                        # "temperature": 0
                     }
                 },
                 {
                     "virtual_key": os.getenv("PORTKEY_MISTRAL_API_KEY"),
                     "override_params": {
                         "model": "codestral-latest",
-                        "max_tokens": 1024,
-                        "temperature": 0
+                        # "max_tokens": 1024,
+                        # "temperature": 0
                     }
                 }
             ]

src/tools/crypto_evm_wallet_toolkit.py

@@ -1,5 +1,4 @@
 import asyncio
-from enum import Enum
 from phi.tools import Toolkit
 from phi.utils.log import logger
 from src.libs.rpc_client import rpc_call
@@ -9,12 +8,13 @@ class CryptoEVMWalletTools(Toolkit):
         super().__init__(name="crypto_evm_wallet_tools")
 
         # Registering methods to make them accessible via the toolkit
-        # self.register(self.get_evm_wallet_address)
         self.register(self.get_supported_evm_chains)
         self.register(self.get_evm_smart_wallet_address)
         self.register(self.get_evm_smart_wallet_balances)
+        self.register(self.get_all_evm_smart_wallet_address)
+        self.register(self.get_all_evm_smart_wallet_balances)
 
-        # Fetching the list of supported EVM chains
+        # Fetch the list of supported EVM chains
         self.chains = self.get_supported_evm_chains()
 
     def get_supported_evm_chains(self) -> list[str]:
@@ -32,35 +32,6 @@ class CryptoEVMWalletTools(Toolkit):
         params = {}
         response = asyncio.run(rpc_call(method_name="getEVMSupportedChains", params=params))
         return f"{response}"
-
-    # def get_evm_wallet_address(self, user_email: str, chain: str, testnet: bool = True) -> str:
-    #     """
-    #     Fetches an EVM wallet address for the given user email and supported chain.
-    #     Creates and returns an EVM wallet address for the given user email and supported chain.
-
-    #     Parameters:
-    #     - user_email (str): The email of the user for whom the wallet is being created.
-    #     - chain (ethereum | binance | base | polygon): The EVM chain for which the wallet is being fetched.
-    #     - testnet (bool, optional): A flag indicating whether the wallet should be on the testnet. Defaults to `True`.
-
-    #     Returns:
-    #     - str: A string representation of the response from the RPC call.
-
-    #     Raises:
-    #     None
-
-    #     Note:
-    #     This method uses asyncio.run() to run the asynchronous RPC call.
-    #     """
-    #     logger.info(f"Creating crypto wallet account for {user_email}")
-
-    #     params = {
-    #         'chain': chain,
-    #         'testnet': testnet,
-    #         'userEmail': user_email,
-    #     }
-    #     response = asyncio.run(rpc_call(method_name="getEVMWallet", params=params))
-    #     return f"{response}"
     
     def get_evm_smart_wallet_address(
         self,
@@ -99,6 +70,34 @@ class CryptoEVMWalletTools(Toolkit):
         response = asyncio.run(rpc_call(method_name="getEVMSmartWallet", params=params))
         return f"{response}"
     
+    def get_all_evm_smart_wallet_address(
+        self,
+        user_email: str,
+        testnet: bool = True
+    ) -> str:
+        """
+        Fetches all EVM smart wallet addresses for a user.
+
+        This method takes a user's email and an optional testnet flag, constructs the
+        necessary parameters, and makes a remote procedure call (RPC) to fetch all EVM
+        smart wallet addresses associated with the user.
+
+        Args:
+            user_email (str): The email of the user whose wallet addresses are to be fetched.
+            testnet (bool, optional): A flag indicating whether to fetch addresses from the testnet. Defaults to True.
+
+        Returns:
+            str: The response from the RPC call containing the wallet addresses.
+        """
+        logger.info(f"Fetching all crypto evm smart wallet addresses for {user_email}")
+
+        params = {
+            'testnet': testnet,
+            'userEmail': user_email,
+        }
+        response = asyncio.run(rpc_call(method_name="getAllEVMSmartWallets", params=params))
+        return f"{response}"
+    
     def get_evm_smart_wallet_balances(
         self,
         address: str,
@@ -134,3 +133,31 @@ class CryptoEVMWalletTools(Toolkit):
         }
         response = asyncio.run(rpc_call(method_name="getEVMSmartWalletAddressBalances", params=params))
         return f"{response}"
+    
+    def get_all_evm_smart_wallet_balances(
+        self,
+        user_email: str,
+        testnet: bool = True
+    ) -> str:
+        """
+        Fetches the balances of all smart EVM wallet addresses for the given user email.
+
+        This method takes a user's email and an optional testnet flag, constructs the
+        necessary parameters, and makes a remote procedure call (RPC) to fetch all EVM
+        smart wallet addresses associated with the user.
+
+        Args:
+            user_email (str): The email of the user whose wallet addresses are to be fetched.
+            testnet (bool, optional): A flag indicating whether to fetch addresses from the testnet. Defaults to True.
+
+        Returns:
+            str: The response from the RPC call containing the wallet addresses.
+        """
+        logger.info(f"Fetching crypto wallet account for {user_email}")
+
+        params = {
+            'testnet': testnet,
+            'userEmail': user_email,
+        }
+        response = asyncio.run(rpc_call(method_name="getAllEVMSmartWalletsAddressBalances", params=params))
+        return f"{response}"