feat: add flock adapter

Author: wosigor

signal-adapters/huma_signals/adapters/flock/__init__.py

@@ -0,0 +1,32 @@
+import datetime
+from typing import Any, ClassVar, List
+
+# import httpx
+# import pandas as pd
+# import pydantic
+import structlog 
+import requests
+
+from huma_signals import models
+from huma_signals.adapters import models as adapter_models
+from huma_signals.settings import settings
+
+logger = structlog.get_logger()
+
+url = "http://35.225.197.208:9999/predict" # flock sandbox api prediction endpoint
+
+class FlockSignals(models.HumaBaseModel):
+    prediction: int
+
+class FlockAdapter(adapter_models.SignalAdapterBase):
+    name: ClassVar[str] = "flock_adapter"
+    required_inputs: ClassVar[List[str]] = ["input_array"]
+    signals: ClassVar[List[str]] = list(FlockSignals.__fields__.keys())
+    
+    async def fetch(self, input_array: dict, *args: Any, **kwargs: Any) -> FlockSignals:
+        headers = {"accept": "application/json"}
+        response = requests.post(url, headers=headers, json={"input_array": input_array})
+        print(response)
+        prediction = response.json()["prediction"]
+        return FlockSignals(prediction=prediction)
+    

signal-adapters/huma_signals/adapters/flock/adapter.py

@@ -9,6 +9,7 @@
 from huma_signals.adapters.stripe import adapter as stripe_adapter
 from huma_signals.adapters.circle import adapter as circle_adapter
 from huma_signals.adapters.spectral import adapter as spectral_adapter
+from huma_signals.adapters.flock import adapter as flock_adapter
 
 ADAPTER_REGISTRY: Dict[str, Type[models.SignalAdapterBase]] = {
     lending_pools_adapter.LendingPoolAdapter.name: lending_pools_adapter.LendingPoolAdapter,
@@ -19,6 +20,7 @@
     stripe_adapter.StripeAdapter.name: stripe_adapter.StripeAdapter,
     circle_adapter.CircleAdapter.name: circle_adapter.CircleAdapter,
     spectral_adapter.SpectralAdapter.name: spectral_adapter.SpectralAdapter,
+    flock_adapter.FlockAdapter.name: flock_adapter.FlockAdapter,
 }