alex/LocalAGI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

s/caption/description

Browse Source 
Signed-off-by: mudler <mudler@localai.io>
This commit is contained in:
mudler
2023-08-26 15:33:01 +02:00
parent d1ae51ae5d
commit 8164190425
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/discord/agent.py
View File

@@ -82,9 +82,9 @@ def ingest(a, agent_actions={}, localagi=None):
def create_image(a, agent_actions={}, localagi=None): def create_image(a, agent_actions={}, localagi=None):
q = json.loads(a) q = json.loads(a)
logger.info(">>> creating image: ") logger.info(">>> creating image: ")
logger.info(q["caption"]) logger.info(q["description"])
size=f"{q['width']}x{q['height']}" size=f"{q['width']}x{q['height']}"
response = openai.Image.create(prompt=q["caption"], n=1, size=size) response = openai.Image.create(prompt=q["description"], n=1, size=size)
image_url = response["data"][0]["url"] image_url = response["data"][0]["url"]
image_name = download_image(image_url) image_name = download_image(image_url)
image_path = f"{PERSISTENT_DIR}{image_name}" image_path = f"{PERSISTENT_DIR}{image_name}"
@@ -252,13 +252,13 @@ agent_actions = {
"generate_picture": { "generate_picture": {
"function": create_image, "function": create_image,
"plannable": True, "plannable": True,
"description": 'For creating a picture, the assistant replies with "generate_picture" and a detailed caption, enhancing it with as much detail as possible.', "description": 'For creating a picture, the assistant replies with "generate_picture" and a detailed description, enhancing it with as much detail as possible.',
"signature": { "signature": {
"name": "generate_picture", "name": "generate_picture",
"parameters": { "parameters": {
"type": "object", "type": "object",
"properties": { "properties": {
"caption": { "description": {
"type": "string", "type": "string",
}, },
"width": { "width": {