Skip to content

Commit

Permalink
feat(TaskProcessing): More task types
Browse files Browse the repository at this point in the history
Signed-off-by: Marcel Klehr <[email protected]>
  • Loading branch information
marcelklehr committed Dec 6, 2024
1 parent 9f0c113 commit f657c7c
Show file tree
Hide file tree
Showing 3 changed files with 271 additions and 0 deletions.
95 changes: 95 additions & 0 deletions lib/public/TaskProcessing/TaskTypes/ContextAgentInteraction.php
Original file line number Diff line number Diff line change
@@ -0,0 +1,95 @@
<?php

declare(strict_types=1);

/**
* SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
* SPDX-License-Identifier: AGPL-3.0-or-later
*/

namespace OCP\TaskProcessing\TaskTypes;

use OCP\IL10N;
use OCP\TaskProcessing\EShapeType;
use OCP\TaskProcessing\ITaskType;
use OCP\TaskProcessing\ShapeDescriptor;

/**
* This is the task processing task type for text reformulation
* @since 31.0.0
*/
class ContextAgentInteraction implements ITaskType {
public const ID = 'core:contextagent:interaction';

public function __construct(
private IL10N $l,
) {
}

/**
* @inheritDoc
*/
public function getName(): string {
return $this->l->t('ContextAgent');
}

/**
* @inheritDoc
*/
public function getDescription(): string {
return $this->l->t('Chat with an agent');
}

/**
* @return string
*/
public function getId(): string {
return self::ID;
}

/**
* @return ShapeDescriptor[]
*/
public function getInputShape(): array {
return [
'input' => new ShapeDescriptor(
$this->l->t('Chat message'),
$this->l->t('A chat message to send to the agent.'),
EShapeType::Text
),
'confirmation' => new ShapeDescriptor(
$this->l->t('Confirmation'),
$this->l->t('Whether to confirm previously requested actions: 0 for denial and 1 for confirmation.'),
EShapeType::Number
),
'conversation_token' => new ShapeDescriptor(
$this->l->t('Conversation token'),
$this->l->t('A token representing the conversation.'),
EShapeType::Text
),
];
}

/**
* @return ShapeDescriptor[]
*/
public function getOutputShape(): array {
return [
'output' => new ShapeDescriptor(
$this->l->t('Generated response'),
$this->l->t('The response from the chat model.'),
EShapeType::Text
),
'conversation_token' => new ShapeDescriptor(
$this->l->t('The new conversation token'),
$this->l->t('Send this along with the next interaction.'),
EShapeType::Text
),
'actions' => new ShapeDescriptor(
$this->l->t('Requested actions by the agent'),
$this->l->t('Actions that the agent would like to carry out in JSON format.'),
EShapeType::Text
),
];
}
}
80 changes: 80 additions & 0 deletions lib/public/TaskProcessing/TaskTypes/TextToTextChangeTone.php
Original file line number Diff line number Diff line change
@@ -0,0 +1,80 @@
<?php

declare(strict_types=1);

/**
* SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
* SPDX-License-Identifier: AGPL-3.0-or-later
*/

namespace OCP\TaskProcessing\TaskTypes;

use OCP\IL10N;
use OCP\TaskProcessing\EShapeType;
use OCP\TaskProcessing\ITaskType;
use OCP\TaskProcessing\ShapeDescriptor;

/**
* This is the task processing task type for text reformulation
* @since 31.0.0
*/
class TextToTextChangeTone implements ITaskType {
public const ID = 'core:text2text:changetone';

public function __construct(
private IL10N $l,
) {
}

/**
* @inheritDoc
*/
public function getName(): string {
return $this->l->t('Change Tone');
}

/**
* @inheritDoc
*/
public function getDescription(): string {
return $this->l->t('Change the tone of a piece of text.');
}

/**
* @return string
*/
public function getId(): string {
return self::ID;
}

/**
* @return ShapeDescriptor[]
*/
public function getInputShape(): array {
return [
'input' => new ShapeDescriptor(
$this->l->t('Input text'),
$this->l->t('Write a text that you want the assistant to rewrite in another tone.'),
EShapeType::Text,
),
'tone' => new ShapeDescriptor(
$this->l->t('Desired tone'),
$this->l->t('In which tone should your text be rewritten?'),
EShapeType::Enum,
),
];
}

/**
* @return ShapeDescriptor[]
*/
public function getOutputShape(): array {
return [
'output' => new ShapeDescriptor(
$this->l->t('Generated response'),
$this->l->t('The rewritten text in the desired tone, written by the assistant:'),
EShapeType::Text
),
];
}
}
96 changes: 96 additions & 0 deletions lib/public/TaskProcessing/TaskTypes/TextToTextChatWithTools.php
Original file line number Diff line number Diff line change
@@ -0,0 +1,96 @@
<?php

declare(strict_types=1);

/**
* SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
* SPDX-License-Identifier: AGPL-3.0-or-later
*/

namespace OCP\TaskProcessing\TaskTypes;

use OCP\IL10N;
use OCP\TaskProcessing\EShapeType;
use OCP\TaskProcessing\ITaskType;
use OCP\TaskProcessing\ShapeDescriptor;

/**
* This is the task processing task type for text reformulation
* @since 31.0.0
*/
class TextToTextChatWithTools implements ITaskType {
public const ID = 'core:text2text:chatwithtools';

public function __construct(
private IL10N $l,
) {
}

/**
* @inheritDoc
*/
public function getName(): string {
return $this->l->t('Chat with tools');
}

/**
* @inheritDoc
*/
public function getDescription(): string {
return $this->l->t('Chat with the language model with tool calling support.');
}

/**
* @return string
*/
public function getId(): string {
return self::ID;
}

/**
* @return ShapeDescriptor[]
*/
public function getInputShape(): array {
return [
'system_prompt' => new ShapeDescriptor(
$this->l->t('System prompt'),
$this->l->t('Define rules and assumptions that the assistant should follow during the conversation.'),
EShapeType::Text
),
'input' => new ShapeDescriptor(
$this->l->t('Chat message'),
$this->l->t('Describe a task that you want the assistant to do or ask a question'),
EShapeType::Text
),
'history' => new ShapeDescriptor(
$this->l->t('Chat history'),
$this->l->t('The history of chat messages before the current message, starting with a message by the user'),
EShapeType::ListOfTexts
),
// See https://platform.openai.com/docs/api-reference/chat/create#chat-create-tools for the format
'tools' => new ShapeDescriptor(
$this->l->t('Available tools'),
$this->l->t('The available tools in JSON format'),
EShapeType::Text
),
];
}

/**
* @return ShapeDescriptor[]
*/
public function getOutputShape(): array {
return [
'output' => new ShapeDescriptor(
$this->l->t('Generated response'),
$this->l->t('The response from the chat model'),
EShapeType::Text
),
'tool_calls' => new ShapeDescriptor(
$this->l->t('Tool calls'),
$this->l->t('Tools call instructions from the model in JSON format'),
EShapeType::Text
),
];
}
}

0 comments on commit f657c7c

Please sign in to comment.