Update anthropic.yaml

anthropic.yaml

@@ -1,9 +1,9 @@
 _id: anthropic
-description: Ask Claude agent
-readme: The second version
 title: Anthropic
-url: https://huggingface.co/PiperMy/Node-Packages/resolve/main/anthropic.yaml
+description: Ask Claude agent
 version: 1
+readme: The first base version
+url: https://huggingface.co/PiperMy/Node-Packages/resolve/main/anthropic.yaml
 nodes:
   ask_claude:
     _id: ask_claude
@@ -33,6 +33,16 @@ nodes:
         default: claude-3-5-sonnet-latest
         enum:
           - claude-3-5-sonnet-latest|Claude 3.5 Sonnet
+      answerFormat:
+        order: 3
+        title: en=Answer format;ru=Формат ответа
+        description: Don't forget add instructions for LLM
+        type: string
+        required: true
+        default: text
+        enum:
+          - text|Text
+          - json|JSON
     outputs:
       answer:
         title: en=Answer;ru=Ответ
@@ -42,17 +52,23 @@ nodes:
         type: json
     package: anthropic
     script: |
-      const Anthropic = require('@anthropic-ai/sdk');
-
-      const ANTHROPIC_API_KEY = env?.variables?.get('ANTHROPIC_API_KEY');
-      if(!ANTHROPIC_API_KEY) {
-          error.fatal('Please, set your API key for Anthropic');
+      export async function costs({ inputs }) {
+          return 0.05;
       }
 
-      const client = new Anthropic({ apiKey: ANTHROPIC_API_KEY });
+      export async function run({ inputs }) {
+
+          const { FatalError, NextNode } = DEFINITIONS;
+          const Anthropic = require('@anthropic-ai/sdk');
 
-      (async () => {
-         const { model, question } = inputs;
+          const ANTHROPIC_API_KEY = env?.variables?.get('ANTHROPIC_API_KEY');
+          if (!ANTHROPIC_API_KEY) {
+              throw new FatalError('Please, set your API key for Anthropic');
+          }
+
+          const client = new Anthropic({ apiKey: ANTHROPIC_API_KEY });
+
+          const { model, question, answerFormat } = inputs;
 
           const message = await client.messages.create({
               max_tokens: 1024,
@@ -60,22 +76,28 @@ nodes:
               model
           });
 
-          const { content: [{type, text: answer}] } = message;
-          log({type});
-          
-          switch(type) {
-              case "text":
-              default:
+          const { content: [{ type, text: answer }] } = message;
+
+          switch (answerFormat) {
+              case 'text':
+                  return NextNode.from({ outputs: { answer }, costs: await costs({ inputs }) });
+              case 'json':
                   try {
                       const json = answer.replace(/^\`\`\`json\s*/ig, '').replace(/\`\`\`\s*$/ig, '');
-                      return next({outputs: { json: JSON.parse(json) }});
-                  } catch(e) {
+                      return NextNode.from({ outputs: { json: JSON.parse(json) }, costs: await costs({ inputs }) });
+                  } catch (e) {
                       log(e);
-                      // plain text
+                      message(`Wrong JSON for question \`\`\`text\n${question}\n\`\`\`\nnanswer from LLM\n\`\`\`text${answer}\n\`\`\``, 'defect');
+                      throw new FatalError("Can't parse JSON asnwer from LLM");
                   }
-                  return next({outputs: { answer }});
+              default:
+                  throw new FatalError(`Wrong answer format ${answerFormat}`);
           }
-      })();
+      }
     source: catalog
     title: en=Ask Claude;ru=Спросить Claude
     version: 1
+    costs: |-
+      (() => {
+          return 0.001;
+      })();