Implement AI workspace and mitigation workflow
This commit is contained in:
@@ -1,14 +1,15 @@
|
||||
from openai import AsyncOpenAI, OpenAI
|
||||
from core.models import Message, ChatSession, AI, Person, Manipulation
|
||||
|
||||
from core.models import AI, ChatSession, Manipulation, Message, Person
|
||||
|
||||
|
||||
async def run_prompt(
|
||||
prompt: list[str],
|
||||
ai: AI,
|
||||
):
|
||||
prompt: list[str],
|
||||
ai: AI,
|
||||
):
|
||||
cast = {"api_key": ai.api_key}
|
||||
if ai.base_url is not None:
|
||||
cast["api_key"] = ai.base_url
|
||||
cast["base_url"] = ai.base_url
|
||||
client = AsyncOpenAI(**cast)
|
||||
response = await client.chat.completions.create(
|
||||
model=ai.model,
|
||||
|
||||
Reference in New Issue
Block a user