forked from React-Group/interstellar_ai
mild changes to the ais and added models to models.tsx
This commit is contained in:
parent
a23257be7f
commit
d83aeb2c74
3 changed files with 21 additions and 7 deletions
9
py/ai.py
9
py/ai.py
|
@ -80,11 +80,6 @@ class AI:
|
|||
message = messages[-1]['content']
|
||||
messages.pop()
|
||||
|
||||
system = None
|
||||
if messages and messages[0]['role'] == 'system':
|
||||
system = messages[0]['content']
|
||||
messages.pop(0)
|
||||
|
||||
for msg in messages:
|
||||
msg['parts'] = msg.pop('content')
|
||||
|
||||
|
@ -97,8 +92,8 @@ class AI:
|
|||
model = genai.GenerativeModel(model)
|
||||
|
||||
chat = model.start_chat(
|
||||
system_instruction=system,
|
||||
history=messages
|
||||
history=messages,
|
||||
|
||||
)
|
||||
|
||||
response = chat.send_message(message, stream=True)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue