clue-api 1.6.0.dev337__tar.gz → 1.6.0.dev339__tar.gz

{clue_api-1.6.0.dev337 → clue_api-1.6.0.dev339}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clue-api
-Version: 1.6.0.dev337
+Version: 1.6.0.dev339
 Summary: Clue distributed enrichment service
 License: MIT
 License-File: LICENSE

{clue_api-1.6.0.dev337 → clue_api-1.6.0.dev339}/clue/models/config.py

@@ -222,6 +222,9 @@ class MongoDB(BaseModel):
     user: str | None = Field(description="Username to use to connect to the MongoDB instance", default=None)
     password: str | None = Field(description="Password to use to connect to the MongoDB instance", default=None)
     database: str = Field(description="The database to use in the mongodb instance", default="clue")
+    auth_source: str | None = Field(
+        description="The database where your user credentials are stored and validated", default="admin"
+    )
     max_retries: int = Field(
         description="Controls the maximum number of retries to use when an initial connection fails", default=2
     )
@@ -253,6 +256,9 @@ class MongoDB(BaseModel):
             "serverSelectionTimeoutMS": self.server_selection_timeout,
         }
 
+        if self.auth_source:
+            params["authSource"] = self.auth_source
+
         if self.user and self.password:
             params["username"] = self.user
             params["password"] = self.password

{clue_api-1.6.0.dev337 → clue_api-1.6.0.dev339}/clue/models/network.py

@@ -388,7 +388,7 @@ class QueryEntry(BaseModel):
         ),
     ] = None
 
-    expiry: datetime | None = Field(description="When should this record expire?", default_factory=generate_expiry)
+    expiry: Timestamp | None = Field(description="When should this record expire?", default_factory=generate_expiry)
 
     model_config = ConfigDict(validate_assignment=True)
 
@@ -566,3 +566,9 @@ class QueryResult(ResultMetadata):
 
 class PluginResponse(BaseModel):
     pass
+
+
+if __name__ == "__main__":
+    import json
+
+    print(json.dumps(QueryEntry.model_json_schema(), indent=2))  # noqa: T201

{clue_api-1.6.0.dev337 → clue_api-1.6.0.dev339}/pyproject.toml

@@ -141,7 +141,7 @@ log_cli_level = "WARN"
 [tool.poetry]
 package-mode = true
 name = "clue-api"
-version = "1.6.0.dev337"
+version = "1.6.0.dev339"
 description = "Clue distributed enrichment service"
 authors = ["Canadian Centre for Cyber Security <contact@cyber.gc.ca>"]
 license = "MIT"