Introduce Expert Agent. Prompt agents from UI.

backend/src/agents/__init__.py

@@ -0,0 +1,2 @@
+from .character import CharacterAgent
+from .expert import ExpertAgent

backend/src/agents/expert.py

@@ -0,0 +1,53 @@
+import re
+import json
+import inspect
+from string import Template
+
+from inference import LlamaMessage, replicate_llama_completion, Models
+from agent import Agent, AgentField
+
+CLASS_DESCRIPTION = """
+Expert AI in any field, powered by Llama 3 70b.
+Will get guidelines, documents, and other advanced features in the future.
+""".strip()
+
+class ExpertAgent(Agent):
+    identifier = "expert"
+    description = CLASS_DESCRIPTION
+    schema_fields = [
+        AgentField('system_prompt', 'string', default="You are an expert AI assistant."),
+    ]
+
+    def __init__(self, app, data):
+        Agent.__init__(self, app, data)
+
+    async def prompt(self, instruction, input=None, **kwargs):
+        # if instruction and input are passed, instruction extends the system prompt
+        # otherwise, use the one passed as the user prompt, with the regular system prompt
+
+        # TODO: if 'history' in kwargs: interpret it as user-agent turns assuming list[str]
+
+        system = self.config['system_prompt']
+        if instruction != None and input != None:
+            system += "\n" + instruction
+
+        prompt = input if input != None else instruction
+        messages = [
+            LlamaMessage("system", system),
+            LlamaMessage("user", prompt)
+        ]
+
+        response = await replicate_llama_completion(self.app, messages, model=Models.llama_70b)
+        return response
+
+    async def received_message(self, message):
+        raise Exception("TODO: expert chat support")
+
+        messages = []
+        response = await replicate_llama_completion(self.app, messages, model=Models.llama_70b)
+        self.app.send_chat_message(
+            chat_id=message.chat_id,
+            sender=self.id,
+            content=response,
+            thread_id=message.thread_id,
+        )

backend/src/app.py

@@ -12,10 +12,10 @@ import replicate
 
 import database as db
 import agent as agent_mod
+import agents as agents_mod
 from server import start_server
 from chat import ChatView, Message, MESSAGE_PAGE_SIZE
 
-from agents.character import CharacterAgent
 from app_mixins import ChatMixin
 
 class App(ChatMixin):
@@ -37,12 +37,13 @@ class App(ChatMixin):
         # `Agent` is the kind 'basic', will be class `BasicAgent` in the future
         self.agent_classes = [
             agent_mod.BasicAgent,
-            CharacterAgent,
+            agents_mod.CharacterAgent,
+            agents_mod.ExpertAgent,
 
-            agent_mod.LlamaAgent,
-            agent_mod.CoderV1Agent,
-            agent_mod.ToolUsingAgent,
-            agent_mod.DebugAgent,
+            #agent_mod.LlamaAgent,
+            #agent_mod.CoderV1Agent,
+            #agent_mod.ToolUsingAgent,
+            #agent_mod.DebugAgent,
         ]
 
     def sql_session(self):
@@ -231,9 +232,9 @@ class App(ChatMixin):
             return fields
         return {}
 
-    async def send_agent_prompt(self, agent_id, message):
+    async def send_agent_prompt(self, agent_id, instruction, input=None, options={}):
         control = self.get_agent_control(agent_id)
-        response = await control.prompt(message)
+        response = await control.prompt(instruction, input, **options)
         return response
 
     async def call_agent_action(self, agent_id, action, args):

backend/src/inference.py

@@ -1,7 +1,6 @@
 import threading
 from stream import Stream
 
-LLAMA3_MODEL = "meta/meta-llama-3-8b-instruct"
 VALID_ROLES = ['user', 'agent', 'system']
 
 class LlamaMessage:

backend/src/server.py

@@ -224,7 +224,16 @@ def start_server(app):
                     await send({'message': "chat thread created", 'threadId': thread_id, 'nonce': nonce})
 
                 elif action == "send-agent-prompt":
-                    response = await app.send_agent_prompt(body['agentId'], body['content'])
+                    if 'instruction' not in body and 'input' not in body:
+                        await send({'error': "instruction or input required"})
+                        continue
+
+                    response = await app.send_agent_prompt(
+                        body['agentId'],
+                        body.get('instruction'),
+                        body.get('input'),
+                        body.get('options', {})
+                    )
                     await send({'message': "propmt sent", 'response': response, 'nonce': nonce})
 
                 elif action == "call-agent-action":

frontend/src/views/Agent.vue

@@ -1,6 +1,6 @@
 <script setup>
   import {
-    ref, watch,
+    ref, watch, reactive,
     onMounted, onUnmounted,
   } from "vue";
 
@@ -60,6 +60,67 @@
     loadAgentData(props.id);
   });
 
+  // === Prompt === //
+  const promptInstructionEnabled = ref(false);
+  const promptInstruction = ref("");
+  const promptInput = ref("");
+
+  const promptInputElem = ref(null);
+  const promptHistory = ref([
+    //{instruction: 'Do something', input: 'With this data', output: "Doing thing with the data"},
+    //{input: 'Do this thing specifically', output: "Ok, I'll do the thing"},
+    //{input: 'Now, do something else', loading: true},
+  ]);
+
+  async function sendPrompt () {
+    let instruction = promptInstruction.value;
+    let input = promptInput.value;
+
+    promptInstruction.value = "";
+    promptInput.value = "";
+
+    let payload = {agentId: agent.value.id};
+    let historyEntry = reactive({loading: true});
+
+    if (promptInstructionEnabled.value) {
+      payload.instruction = instruction;
+      payload.input = input;
+
+      historyEntry.instruction = instruction;
+      historyEntry.input = input;
+    } else {
+      // uhm
+      payload.instruction = input;
+      historyEntry.input = input;
+    }
+
+    promptHistory.value.push(historyEntry);
+    let res = await backend.sendAction("send-agent-prompt", payload);
+
+    if (res.error) {
+      historyEntry.error = res.error;
+    } else {
+      historyEntry.output = res.response;
+    }
+    historyEntry.loading = false;
+  }
+
+  function promptKeyDown (event, stage) {
+    if (event.key == "Enter" && event.ctrlKey) {
+      event.preventDefault();
+      switch (stage) {
+        case 1:
+          promptInputElem.value.focus();
+          break;
+        case 2:
+          sendPrompt();
+          break;
+      }
+    }
+  }
+
+  // === Configuration === //
+
   function updateIdentifier() {
     const value = prompt(`Enter a new identifier for ${agent.value.name}`);
     if (value) {
@@ -129,22 +190,55 @@
   <main>
     <h1>{{ agent.name || `Agent-${id}` }}</h1>
     <section class="top-section">
-      <div>
-        <span class="label">Agent ID</span>
-        <span class="value">{{id}}</span>
-      </div>
-      <div>
-        <span class="label">Identifier</span>
-        <span class="value">{{agent.agentname}}</span>
-        <button @click="updateIdentifier">Edit</button>
-      </div>
-      <div>
-        <button @click="updateName">Change Name</button>
+      <p style="white-space: pre-wrap; padding-bottom: 1em; border-bottom: 1px solid var(--color-border)">{{ kindInfo.description }}</p>
+
+      <div class="basic-config" style="padding-top: 1em;">
+        <div>
+          <span class="label">Agent ID</span>
+          <span class="value">{{id}}</span>
+        </div>
+        <div>
+          <span class="label">Identifier</span>
+          <span class="value">{{agent.agentname}}</span>
+          <button @click="updateIdentifier">Edit</button>
+        </div>
+        <div>
+          <button @click="updateName">Change Name</button>
+        </div>
       </div>
     </section>
     <section>
-      <p style="padding-bottom: 1em; border-bottom: 1px solid var(--color-border)">{{ kindInfo.description }}</p>
+      <h3>Prompt</h3>
 
+      <div>
+        <span class="label">Enable Instruction/Input</span>
+        <input type="checkbox" v-model="promptInstructionEnabled" />
+      </div>
+      <textarea v-if="promptInstructionEnabled" v-model="promptInstruction" class="config-input" @keydown="promptKeyDown($event, 1)" />
+      <textarea v-model="promptInput" class="config-input" ref="promptInputElem" @keydown="promptKeyDown($event, 2)" />
+      <div>
+        <button @click="sendPrompt()">Send Prompt</button>
+      </div>
+
+      <div v-for="entry in promptHistory" class="prompt-history-entry">
+        <div v-if="entry.instruction">
+          <div class="label">Instruction</div>
+          <pre>{{entry.instruction}}</pre>
+        </div>
+        <div>
+          <div class="label" v-if="entry.instruction">Input</div>
+          <div class="label" v-else>Prompt</div>
+          <pre>{{entry.input}}</pre>
+        </div>
+        <div>
+          <div class="label">Output</div>
+          <div v-if="entry.loading">Loading...</div>
+          <pre v-if="entry.output">{{entry.output}}</pre>
+        </div>
+      </div>
+
+    </section>
+    <section>
       <div v-for="(configField, index) in configFields" :key="index">
         <span class="label">{{ configField.name }}</span>
         <input type="checkbox" v-model="configField.enabled" />
@@ -165,13 +259,21 @@
     border-radius: var(--border-radius);
   }
 
-  .top-section {
-    display: flex;
-    flex-direction: row;
+  .basic-config > div {
+    margin-right: 2em;
+    display: inline-block;
   }
 
-  .top-section > div {
-    margin-right: 2em;
+  .prompt-history-entry {
+    border-top: 1px dashed var(--color-border);
+    padding: 1em 0;
+  }
+  
+  .prompt-history-entry pre {
+    border: 1px solid var(--color-border);
+    border-radius: var(--border-radius);
+    padding: 0.5em;
+    white-space: pre-wrap;
   }
 
   .label {