Initial commit

Author: franmaranchello

Diff for: .gitignore

@@ -0,0 +1,11 @@
+node_modules
+.env
+coverage
+coverage.json
+yarn.lock
+keys
+
+#Build outputs
+cache
+artifacts
+

Diff for: askZoidberg.js

@@ -0,0 +1,19 @@
+const sendToClaude = require("./sendToClaude");
+
+async function askZoidberg(userQuestion) {
+  try {
+    // Step 2: Build Prompt
+    let prompt = `This is the user request: ${userQuestion}.`;
+
+    // Step 3: Send Data to the Antrhopic API
+    const code = await sendToClaude(prompt);
+
+    // Step 4: Return code
+    return code;
+  } catch (error) {
+    console.error("An error occurred:", error);
+    throw error;
+  }
+}
+
+module.exports = askZoidberg;

Diff for: index.js

@@ -0,0 +1,25 @@
+require("dotenv").config();
+const askZoidberg = require("./askZoidberg.js");
+
+exports.zoidbergAssistant = async (req, res) => {
+  const apiKey = req.headers["x-api-key"] || req.query.api_key;
+
+  if (!apiKey || apiKey !== process.env.ENDPOINT_API_KEY) {
+    res.status(401).send({ error: "Unauthorized" });
+    return;
+  }
+
+  try {
+    const userInstruction =
+      req.query.instruction ||
+      res.status(400).send({
+        error:
+          "Please provide an instruction for Zoidberg in your query parameters.",
+      });
+    const answer = await askZoidberg(userInstruction);
+    res.status(200).send({ answer });
+  } catch (error) {
+    console.error("Error:", error);
+    res.status(500).send({ error: "An error occurred" });
+  }
+};

Diff for: package.json

@@ -0,0 +1,17 @@
+{
+  "name": "aecworks-chatbot",
+  "version": "1.0.0",
+  "dependencies": {
+    "@anthropic-ai/sdk": "^0.19.0",
+    "@google-cloud/bigquery": "^7.5.1",
+    "@google-cloud/functions-framework": "^3.0.0",
+    "@types/node": "^20.11.30",
+    "express": "^4.19.2"
+  },
+  "scripts": {
+    "start": "functions-framework --target=aecWorksChatbot"
+  },
+  "devDependencies": {
+    "dotenv": "^16.4.5"
+  }
+}

Diff for: sendToClaude.js

@@ -0,0 +1,25 @@
+const { Anthropic } = require("@anthropic-ai/sdk");
+require("dotenv").config();
+
+async function sendToClaude(prompt) {
+  const anthropic = new Anthropic({ apiKey: process.env.ANTHROPIC_API_KEY });
+
+  try {
+    const response = await anthropic.messages.create({
+      model: process.env.ANTHROPIC_MODEL,
+      max_tokens: 4096,
+      temperature: 0,
+      system:
+        "You're Zoidberg, an expert in AEC technology and BIM development that helps with Revit requests. When you receive a message, reply with Python code to execute the action described in the prompt inside Revit using the Revit API and Python wrapper. Don't add anything that isn't code, not a single character.",
+      messages: [{ role: "user", content: prompt }],
+    });
+    console.log(response);
+
+    return response.content[0].text;
+  } catch (error) {
+    console.error("Error sending message to Claude:", error);
+    throw error;
+  }
+}
+
+module.exports = sendToClaude;