import httpx
|
|
from app.config.config import settings
|
from app.utils.rsa_crypto import RagflowCrypto
|
|
|
class RagflowService:
|
def __init__(self, base_url: str):
|
self.base_url = base_url
|
|
async def register(self, username: str, password: str):
|
password = RagflowCrypto(settings.PUBLIC_KEY, settings.PRIVATE_KEY).encrypt(password)
|
async with httpx.AsyncClient() as client:
|
response = await client.post(
|
f"{self.base_url}/v1/user/register",
|
json={"nickname": username, "email": f"{username}@example.com", "password": password},
|
headers={'Content-Type': 'application/json'}
|
)
|
if response.status_code != 200:
|
raise Exception(f"Ragflow registration failed: {response.text}")
|
|
async def login(self, username: str, password: str) -> str:
|
password = RagflowCrypto(settings.PUBLIC_KEY, settings.PRIVATE_KEY).encrypt(password)
|
async with httpx.AsyncClient() as client:
|
response = await client.post(
|
f"{self.base_url}/v1/user/login",
|
json={"email": f"{username}@example.com", "password": password},
|
headers={'Content-Type': 'application/json'}
|
)
|
if response.status_code != 200:
|
raise Exception(f"Ragflow login failed: {response.text}")
|
return response.json().get('data', {}).get('access_token')
|
