metaai-sdk 2.3.6__tar.gz → 3.0.0__tar.gz

{metaai_sdk-2.3.6 → metaai_sdk-3.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: metaai-sdk
-Version: 2.3.6
+Version: 3.0.0
 Summary: Feature-rich Python SDK for Meta AI - Chat, Image & Video Generation powered by Llama 3
 Author-email: Ashiq Hussain Mir <imseldrith@gmail.com>
 License-Expression: MIT

{metaai_sdk-2.3.6 → metaai_sdk-3.0.0}/src/metaai_api/api_server.py

@@ -107,12 +107,14 @@ class ImageRequest(BaseModel):
     new_conversation: bool = False
     media_ids: Optional[list] = None
     attachment_metadata: Optional[dict] = None  # {'file_size': int, 'mime_type': str}
+    orientation: Optional[str] = None  # 'VERTICAL', 'LANDSCAPE' (not HORIZONTAL), or 'SQUARE'
 
 
 class VideoRequest(BaseModel):
     prompt: str
     media_ids: Optional[list] = None
     attachment_metadata: Optional[dict] = None  # {'file_size': int, 'mime_type': str}
+    orientation: Optional[str] = None  # 'VERTICAL', 'LANDSCAPE', or 'SQUARE'
     wait_before_poll: int = Field(10, ge=0, le=60)
     max_attempts: int = Field(30, ge=1, le=60)
     wait_seconds: int = Field(5, ge=1, le=30)
@@ -206,7 +208,14 @@ async def chat(body: ChatRequest, cookies: Dict[str, str] = Depends(get_cookies)
         raise HTTPException(status_code=400, detail="Streaming not supported via HTTP JSON; set stream=false")
     ai = MetaAI(cookies=cookies, proxy=_get_proxies())
     try:
-        return cast(Dict[str, Any], ai.prompt(body.message, stream=False, new_conversation=body.new_conversation, media_ids=body.media_ids, attachment_metadata=body.attachment_metadata))
+        return cast(Dict[str, Any], await run_in_threadpool(
+            ai.prompt,
+            body.message,
+            stream=False,
+            new_conversation=body.new_conversation,
+            media_ids=body.media_ids,
+            attachment_metadata=body.attachment_metadata
+        ))
     except Exception as exc:  # noqa: BLE001
         await cache.refresh_after_error()
         raise HTTPException(status_code=502, detail=str(exc)) from exc
@@ -218,13 +227,15 @@ async def image(body: ImageRequest, cookies: Dict[str, str] = Depends(get_cookie
     try:
         # Automatically prepend "generate image of" to the prompt
         prompt = f"generate image of {body.prompt}" if not body.prompt.lower().startswith(("generate image", "create image")) else body.prompt
-        return cast(Dict[str, Any], ai.prompt(
-            prompt, 
-            stream=False, 
-            new_conversation=body.new_conversation, 
-            media_ids=body.media_ids, 
+        return cast(Dict[str, Any], await run_in_threadpool(
+            ai.prompt,
+            prompt,
+            stream=False,
+            new_conversation=body.new_conversation,
+            media_ids=body.media_ids,
             attachment_metadata=body.attachment_metadata,
-            is_image_generation=True  # Image generation uses DISCOVER entrypoint
+            is_image_generation=True,  # Image generation uses DISCOVER entrypoint
+            orientation=body.orientation  # Support VERTICAL, HORIZONTAL, SQUARE
         ))
     except Exception as exc:  # noqa: BLE001
         await cache.refresh_after_error()
@@ -233,6 +244,10 @@ async def image(body: ImageRequest, cookies: Dict[str, str] = Depends(get_cookie
 
 @app.post("/video")
 async def video(body: VideoRequest, cookies: Dict[str, str] = Depends(get_cookies)) -> Dict[str, Any]:
+    # Force refresh cookies before video generation for best results
+    await cache.refresh_if_needed(force=True)
+    cookies = await cache.snapshot()
+    
     ai = MetaAI(cookies=cookies, proxy=_get_proxies())
     try:
         # Automatically prepend "generate a video" to the prompt
@@ -242,6 +257,7 @@ async def video(body: VideoRequest, cookies: Dict[str, str] = Depends(get_cookie
             prompt,
             body.media_ids,
             body.attachment_metadata,
+            body.orientation,
             body.wait_before_poll,
             body.max_attempts,
             body.wait_seconds,
@@ -254,6 +270,10 @@ async def video(body: VideoRequest, cookies: Dict[str, str] = Depends(get_cookie
 
 @app.post("/video/async")
 async def video_async(body: VideoRequest, cookies: Dict[str, str] = Depends(get_cookies)) -> Dict[str, str]:
+    # Force refresh cookies before video generation
+    await cache.refresh_if_needed(force=True)
+    cookies = await cache.snapshot()
+    
     job = await jobs.create()
     asyncio.create_task(_run_video_job(job.job_id, body, cookies))
     return {"job_id": job.job_id, "status": "pending"}
@@ -322,6 +342,7 @@ async def _run_video_job(job_id: str, body: VideoRequest, cookies: Dict[str, str
             body.prompt,
             body.media_ids,
             body.attachment_metadata,
+            body.orientation,
             body.wait_before_poll,
             body.max_attempts,
             body.wait_seconds,

{metaai_sdk-2.3.6 → metaai_sdk-3.0.0}/src/metaai_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: metaai-sdk
-Version: 2.3.6
+Version: 3.0.0
 Summary: Feature-rich Python SDK for Meta AI - Chat, Image & Video Generation powered by Llama 3
 Author-email: Ashiq Hussain Mir <imseldrith@gmail.com>
 License-Expression: MIT

{metaai_sdk-2.3.6 → metaai_sdk-3.0.0}/src/metaai_sdk.egg-info/SOURCES.txt

@@ -22,8 +22,13 @@ meta_ai_debug.log
 pyproject.toml
 quick_video_test.py
 requirements.txt
+test_api_quick.py
+test_comprehensive.py
 test_endpoints.py
 test_image_upload.py
+test_orientation.py
+test_orientation_all.py
+test_video_orientation.py
 examples/complete_workflow_example.py
 examples/image_upload_example.py
 examples/image_workflow_complete.py

metaai_sdk-3.0.0/test_api_quick.py

@@ -0,0 +1,179 @@
+"""
+Quick API test with timeout handling
+"""
+
+import requests
+import json
+
+API_URL = "http://localhost:8000"
+TIMEOUT = 120  # 2 minutes timeout for each request
+
+def test_health():
+    """Test health check endpoint"""
+    print("\n" + "="*70)
+    print("TEST: Health Check")
+    print("="*70)
+    try:
+        response = requests.get(f"{API_URL}/healthz", timeout=5)
+        print(f"✓ Status: {response.status_code}")
+        print(f"  Response: {response.json()}")
+        return True
+    except Exception as e:
+        print(f"✗ Error: {e}")
+        return False
+
+def test_chat():
+    """Test basic chat endpoint"""
+    print("\n" + "="*70)
+    print("TEST: Chat Endpoint (Simple)")
+    print("="*70)
+    try:
+        payload = {
+            "message": "What is 2+2?",
+            "stream": False,
+            "new_conversation": True
+        }
+        print(f"Request: {payload}")
+        response = requests.post(
+            f"{API_URL}/chat",
+            headers={"Content-Type": "application/json"},
+            json=payload,
+            timeout=TIMEOUT
+        )
+        print(f"✓ Status: {response.status_code}")
+        if response.status_code == 200:
+            result = response.json()
+            print(f"  Message: {result.get('message', 'No message')[:200]}")
+            return True
+        else:
+            print(f"  Error: {response.text}")
+            return False
+    except requests.Timeout:
+        print(f"✗ Timeout after {TIMEOUT}s")
+        return False
+    except Exception as e:
+        print(f"✗ Error: {e}")
+        return False
+
+def test_upload():
+    """Test image upload endpoint"""
+    print("\n" + "="*70)
+    print("TEST: Image Upload")
+    print("="*70)
+    
+    import os
+    from pathlib import Path
+    
+    # Find an image to upload
+    image_path = r"C:\Users\spike\Downloads\meta-ai-api-main\ChatGPT Image Jan 14, 2026, 06_59_02 PM.png"
+    
+    if not os.path.exists(image_path):
+        print(f"✗ Image not found: {image_path}")
+        return False, None
+    
+    try:
+        with open(image_path, 'rb') as f:
+            files = {'file': f}
+            response = requests.post(
+                f"{API_URL}/upload",
+                files=files,
+                timeout=TIMEOUT
+            )
+        
+        print(f"✓ Status: {response.status_code}")
+        if response.status_code == 200:
+            result = response.json()
+            media_id = result.get('media_id')
+            print(f"  Media ID: {media_id}")
+            print(f"  File Size: {result.get('file_size')} bytes")
+            print(f"  MIME Type: {result.get('mime_type')}")
+            metadata = {
+                'file_size': result.get('file_size'),
+                'mime_type': result.get('mime_type')
+            }
+            return True, (media_id, metadata)
+        else:
+            print(f"  Error: {response.text}")
+            return False, None
+    except Exception as e:
+        print(f"✗ Error: {e}")
+        return False, None
+
+def test_image_generation():
+    """Test image generation endpoint"""
+    print("\n" + "="*70)
+    print("TEST: Image Generation")
+    print("="*70)
+    try:
+        payload = {
+            "prompt": "a cute cat wearing sunglasses",
+            "new_conversation": True
+        }
+        print(f"Request: {payload}")
+        response = requests.post(
+            f"{API_URL}/image",
+            headers={"Content-Type": "application/json"},
+            json=payload,
+            timeout=TIMEOUT
+        )
+        print(f"✓ Status: {response.status_code}")
+        if response.status_code == 200:
+            result = response.json()
+            print(f"  Message: {result.get('message', 'No message')[:200]}")
+            media = result.get('media', [])
+            print(f"  Media count: {len(media)}")
+            for i, m in enumerate(media[:2], 1):
+                print(f"    {i}. {m.get('url', 'No URL')[:80]}...")
+            return True
+        else:
+            print(f"  Error: {response.text}")
+            return False
+    except requests.Timeout:
+        print(f"✗ Timeout after {TIMEOUT}s")
+        return False
+    except Exception as e:
+        print(f"✗ Error: {e}")
+        return False
+
+def main():
+    """Run all tests"""
+    print("\n" + "+"*70)
+    print("  API QUICK TEST SUITE")
+    print("+"*70)
+    print(f"\nAPI URL: {API_URL}")
+    print(f"Timeout: {TIMEOUT}s per request\n")
+    
+    results = {}
+    
+    # Test 1: Health
+    results['health'] = test_health()
+    
+    # Test 2: Chat
+    results['chat'] = test_chat()
+    
+    # Test 3: Upload
+    upload_success, upload_data = test_upload()
+    results['upload'] = upload_success
+    
+    # Test 4: Image Generation
+    results['image'] = test_image_generation()
+    
+    # Summary
+    print("\n" + "="*70)
+    print("SUMMARY")
+    print("="*70)
+    for test_name, passed in results.items():
+        status = "✓ PASS" if passed else "✗ FAIL"
+        print(f"{test_name.ljust(20)}: {status}")
+    
+    total = len(results)
+    passed = sum(results.values())
+    print(f"\nTotal: {passed}/{total} tests passed")
+    
+    if passed == total:
+        print("\n✓ All tests passed!")
+    else:
+        print(f"\n✗ {total - passed} test(s) failed")
+
+if __name__ == "__main__":
+    main()