rafael1994s commited on
Commit
fc0b088
·
verified ·
1 Parent(s): 4390bfe

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -5
main.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- # Это должно стоять ДО импорта g4f
3
  os.environ['G4F_DATA_DIRECTORY'] = '/tmp/har_and_cookies'
4
 
5
  from fastapi import FastAPI
@@ -9,16 +9,16 @@ app = FastAPI()
9
 
10
  @app.get("/")
11
  def home():
12
- return {"status": "Working"}
13
 
14
  @app.get("/ask")
15
  async def ask(prompt: str):
16
  try:
17
- # Используем Blackbox, он самый "некапризный" к папкам
 
18
  response = g4f.ChatCompletion.create(
19
- model="gpt-4o",
20
  messages=[{"role": "user", "content": prompt}],
21
- provider=g4f.Provider.Blackbox,
22
  )
23
  return {"answer": response}
24
  except Exception as e:
 
1
  import os
2
+ # Указываем путь к папке, которую мы разрешили в Dockerfile
3
  os.environ['G4F_DATA_DIRECTORY'] = '/tmp/har_and_cookies'
4
 
5
  from fastapi import FastAPI
 
9
 
10
  @app.get("/")
11
  def home():
12
+ return {"status": "API is running"}
13
 
14
  @app.get("/ask")
15
  async def ask(prompt: str):
16
  try:
17
+ # Убираем явное указание провайдера.
18
+ # Библиотека сама переберет все варианты (Blackbox, DuckDuckGo и др.)
19
  response = g4f.ChatCompletion.create(
20
+ model=g4f.models.default, # Автовыбор модели
21
  messages=[{"role": "user", "content": prompt}],
 
22
  )
23
  return {"answer": response}
24
  except Exception as e: