File size: 819 Bytes
6f79314
fc0b088
4390bfe
6f79314
3cb3bbd
 
 
 
 
 
 
fc0b088
3cb3bbd
 
 
 
fc0b088
 
3cb3bbd
fc0b088
3cb3bbd
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import os
# Указываем путь к папке, которую мы разрешили в Dockerfile
os.environ['G4F_DATA_DIRECTORY'] = '/tmp/har_and_cookies'

from fastapi import FastAPI
import g4f

app = FastAPI()

@app.get("/")
def home():
    return {"status": "API is running"}

@app.get("/ask")
async def ask(prompt: str):
    try:
        # Убираем явное указание провайдера. 
        # Библиотека сама переберет все варианты (Blackbox, DuckDuckGo и др.)
        response = g4f.ChatCompletion.create(
            model=g4f.models.default, # Автовыбор модели
            messages=[{"role": "user", "content": prompt}],
        )
        return {"answer": response}
    except Exception as e:
        return {"error": str(e)}