prediction-market-agent-tooling 0.32.0__tar.gz → 0.33.0__tar.gz

{prediction_market_agent_tooling-0.32.0 → prediction_market_agent_tooling-0.33.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.32.0
+Version: 0.33.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12

prediction_market_agent_tooling-0.33.0/prediction_market_agent_tooling/tools/streamlit_user_login.py

@@ -0,0 +1,82 @@
+from enum import Enum
+
+import streamlit as st
+from pydantic import BaseModel, SecretStr
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+from prediction_market_agent_tooling.loggers import logger
+
+
+class LoggedInUser(BaseModel):
+    email: str
+    password: SecretStr
+
+
+class LoginSettings(BaseSettings):
+    model_config = SettingsConfigDict(
+        env_file=".env", env_file_encoding="utf-8", extra="ignore"
+    )
+    free_for_everyone: bool = False
+    free_access_codes: list[SecretStr] = []
+    users: list[LoggedInUser] = []
+
+
+class LoggedEnum(str, Enum):
+    SELF_PAYING = "self_paying"
+    FREE_ACCESS = "free_access"
+    USER_LOGGED_IN = "user_logged_in"
+
+
+def find_logged_in_user(email: str, password: SecretStr) -> LoggedInUser | None:
+    login_settings = LoginSettings()
+
+    for user in login_settings.users:
+        if (
+            user.email == email
+            and user.password.get_secret_value() == password.get_secret_value()
+        ):
+            return user
+
+    return None
+
+
+def streamlit_login() -> tuple[LoggedEnum, LoggedInUser | None]:
+    login_settings = LoginSettings()
+
+    if login_settings.free_for_everyone:
+        logger.info("Free access for everyone!")
+        return LoggedEnum.FREE_ACCESS, None
+
+    if (
+        free_access_code := st.query_params.get("free_access_code")
+    ) is not None and free_access_code in [
+        x.get_secret_value() for x in login_settings.free_access_codes
+    ]:
+        logger.info(f"Using free access code: {free_access_code}.")
+        return LoggedEnum.FREE_ACCESS, None
+
+    # TODO: Because we are initializing APIKeys (based on environment variables) in random places in the code, the user can not provide their own access keys, such as OPENAI_API_KEY.
+    # Doing that would require to change the environment variables in the code, but Streamlit is using 1 thread per session, and the environment variables are shared between threads.
+    # So other users would get the secrets provided by the first user!
+    # Another option is to refactor the code, such that APIKeys are initialized in a single place, somewhere in the agent's initialisation and then, we can initialize them here based on the user's input and just provide them  to the agent.
+    # After that's done, we should also have unique api keys per registered user.
+    self_paying = False
+    # self_paying = st.checkbox("I will provide my own access keys", value=False)
+
+    if not self_paying:
+        email = st.text_input("Email")
+        password = SecretStr(st.text_input("Password", type="password"))
+        logged_user = find_logged_in_user(email, password)
+
+        if logged_user is not None:
+            logger.info(f"Logged in as {email}.")
+            return LoggedEnum.USER_LOGGED_IN, logged_user
+
+        else:
+            st.error("Invalid email or password.")
+            st.stop()
+
+    else:
+        raise NotImplementedError(
+            "Should not happen. Self-paying is not implemented yet. See the comment above."
+        )

{prediction_market_agent_tooling-0.32.0 → prediction_market_agent_tooling-0.33.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.32.0"
+version = "0.33.0"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"