cycls 0.0.2.24__tar.gz → 0.0.2.30__tar.gz

{cycls-0.0.2.24 → cycls-0.0.2.30}/PKG-INFO

@@ -1,19 +1,20 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: cycls
-Version: 0.0.2.24
+Version: 0.0.2.30
 Summary: Cycls SDK
-Author: Mohammed Jamal
+Author: Mohammed J. AlRujayi
 Author-email: mj@cycls.com
-Requires-Python: >=3.8,<4.0
+Requires-Python: >=3.9,<4.0
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: asyncssh (>=2.14.2,<3.0.0)
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: fastapi (>=0.111.0,<0.112.0)
 Requires-Dist: httpx (>=0.27.0,<0.28.0)
+Requires-Dist: jwt (>=1.4.0,<2.0.0)
+Requires-Dist: modal (>=1.1.0,<2.0.0)
 Description-Content-Type: text/markdown
 
 <p align="center">

cycls-0.0.2.30/cycls/__init__.py

@@ -0,0 +1 @@
+from .cycls import Agent

cycls-0.0.2.30/cycls/cycls.py

@@ -0,0 +1,162 @@
+import json, time, modal, inspect, uvicorn
+from modal.runner import run_app
+
+import importlib.resources
+theme_path = importlib.resources.files('cycls').joinpath('theme')
+
+async def openai_encoder(stream): # clean up the meta data / new API?
+    async for message in stream:
+        payload = {"id": "chatcmpl-123",
+                   "object": "chat.completion.chunk",
+                   "created": 1728083325,
+                   "model": "model-1-2025-01-01",
+                   "system_fingerprint": "fp_123456",
+                   "choices": [{"delta": {"content": message}}]}
+        if message:
+            yield f"data: {json.dumps(payload)}\n\n"
+    yield "data: [DONE]\n\n"
+
+test_auth_public_key = """
+-----BEGIN PUBLIC KEY-----
+MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAyDudrDtQ5irw6hPWf2rw
+FvNAFWeOouOO3XNWVQrjXCZfegiLYkL4cJdm4eqIuMdFHGnXU+gWT5P0EkLIkbtE
+zpqDb5Wp27WpSRb5lqJehpU7FE+oQuovCwR9m5gYXP5rfM+CQ7ZPw/CcOQPtOB5G
+0UijBhmYqws3SFp1Rk1uFed1F/esspt6Ifq2uDSHESleylqTKUCQiBa++z4wllcV
+PbNiooLRpsF0kGljP2dXXy/ViF7q9Cblgl+FdrqtGfHD+DHJuOSYcPnRa0IHZYS4
+r5i9C2lejVrEDqgJk5IbmQgez0wmEG4ynAxiDLvfdtvrd27PyBI75FsyLER/ydBH
+WwIDAQAB
+-----END PUBLIC KEY-----
+"""
+
+live_auth_public_key = """
+-----BEGIN PUBLIC KEY-----
+MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAorfL7XyxrLG/X+Kq9ImY
+oSQ+Y3PY5qi8t8R4urY9u4ADJ48j9LkmFz8ALbubQkl3IByDDuVbka49m8id9isy
+F9ZJErsZzzlYztrgI5Sg4R6OJXcNWLqh/tzutMWJFOrE3LnHXpeyQMo/6qAd59Dx
+sNqzGxBTGPV1BZvpfhp/TT/sjgbPQWHS4PMpKD4vZLKXeTNJ913fMTUoFAIaL0sT
+EhoeLUwvIuhLx4UYTmjO/sa+fS6mdghjddOkjSS/AWr/K8mN3IXDImGqh83L7/P0
+RCru4Hvarm0qPIhfwEFfWhKFXONMj3x2fT4MM1Uw1H7qKTER2MtOjmdchKNX7x9b
+XwIDAQAB
+-----END PUBLIC KEY-----
+"""
+
+def web(func, front_end_path="", prod=False, org=None, api_token=None, header="", intro="", auth=True): # API auth
+    print(front_end_path)
+    from fastapi import FastAPI, Request, HTTPException, status, Depends
+    from fastapi.responses import StreamingResponse , HTMLResponse
+    from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
+    import jwt
+    from pydantic import BaseModel, EmailStr
+    from typing import List, Optional
+    from fastapi.templating import Jinja2Templates
+    from fastapi.staticfiles import StaticFiles
+
+    class User(BaseModel):
+        id: str
+        name: str
+        email: EmailStr
+        org: Optional[str] = None
+        plans: List[str] = []
+
+    class Context(BaseModel):
+        messages: List[dict]
+        user: Optional[User] = None
+
+    app = FastAPI()
+    bearer_scheme = HTTPBearer()
+
+    def validate(bearer: HTTPAuthorizationCredentials = Depends(bearer_scheme)):
+        # if api_token and api_token==""
+        try:
+            public_key = live_auth_public_key if prod else test_auth_public_key
+            decoded = jwt.decode(bearer.credentials, public_key, algorithms=["RS256"])
+            # print(decoded)
+            return {"type": "user", 
+                    "user": {"id": decoded.get("id"), "name": decoded.get("name"), "email": decoded.get("email"), "org": decoded.get("org"),
+                             "plans": decoded.get("public").get("plans", [])}}
+        except:
+            raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid credentials", headers={"WWW-Authenticate": "Bearer"})
+    
+    @app.post("/")
+    @app.post("/chat/completions")
+    async def back(request: Request, jwt: Optional[dict] = Depends(validate) if auth else None):
+        data = await request.json()
+        messages = data.get("messages")
+        user_data = jwt.get("user") if jwt else None
+        context = Context(messages = messages, user = User(**user_data) if user_data else None)
+        stream = await func(context) if inspect.iscoroutinefunction(func) else func(context)
+        if request.url.path == "/chat/completions":
+            stream = openai_encoder(stream)
+        return StreamingResponse(stream, media_type="text/event-stream")
+
+    templates = Jinja2Templates(directory=front_end_path)
+    @app.get("/", response_class=HTMLResponse)
+    async def front(request: Request):
+        return templates.TemplateResponse("index.html", {
+                "request": request, "header": header, "intro": intro, "prod": prod, "auth": auth, "org": org,
+                "pk_live": "pk_live_Y2xlcmsuY3ljbHMuY29tJA", "pk_test": "pk_test_c2VsZWN0LXNsb3RoLTU4LmNsZXJrLmFjY291bnRzLmRldiQ"
+            })
+    app.mount("/", StaticFiles(directory=front_end_path, html=False))
+    return app
+
+class Agent:
+    def __init__(self, front_end=theme_path, organization=None, api_token=None, pip=[], apt=[], copy=[], production=False, keys=["",""]):
+        self.prod, self.org, self.api_token = production, organization, api_token
+        self.front_end = front_end
+        self.registered_functions = []
+        self.client = modal.Client.from_credentials(*keys)
+        image = (modal.Image.debian_slim()
+                            .pip_install("fastapi[standard]", "pyjwt", "cryptography", *pip)
+                            .apt_install(*apt)
+                            .add_local_dir(front_end, "/root/public")
+                            .add_local_python_source("cycls"))        
+        for item in copy:
+            image = image.add_local_file(item, f"/root/{item}") if "." in item else image.add_local_dir(item, f'/root/{item}')
+        self.app = modal.App("development", image=image)
+
+    def __call__(self, name="", header="", intro="", domain=None, auth=False):
+        def decorator(f):
+            self.registered_functions.append({
+                "func": f,
+                "config": ["public", self.prod, self.org, self.api_token, header, intro, auth],
+                "name": name,
+                "domain": domain or f"{name}.cycls.ai",
+            })
+            return f
+        return decorator
+
+    def run(self, port=8000):
+        if not self.registered_functions:
+            return print("Error: No @agent decorated function found.")
+        
+        i = self.registered_functions[0]
+        if len(self.registered_functions) > 1:
+            print(f"⚠️  Warning: Multiple agents found. Running '{i['name']}'.")
+        print(f"🚀 Starting local server at http://127.0.0.1:{port}")
+        i["config"][0] = self.front_end
+        uvicorn.run(web(i["func"], *i["config"]), host="127.0.0.1", port=port)
+        return
+
+    def push(self): # local / prod?
+        if not self.registered_functions:
+            return print("Error: No @agent decorated function found.")
+
+        for i in self.registered_functions:
+            self.app.function(serialized=True, name=i["name"])(
+                modal.asgi_app(label=i["name"], custom_domains=[i["domain"]])
+                (lambda: web(i["func"], *i["config"]))
+            )
+        if self.prod:
+            for i in self.registered_functions:
+                print(f"✅ Deployed to ⇒ https://{i['domain']}")
+            self.app.deploy(client=self.client, name=self.registered_functions[0]["name"])
+            return
+        else:
+            with modal.enable_output():
+                run_app(app=self.app, client=self.client)
+                print(" Modal development server is running. Press Ctrl+C to stop.")
+                with modal.enable_output(), run_app(app=self.app, client=self.client): 
+                    while True: time.sleep(10)
+
+# poetry run python agent.py
+# poetry publish --build