Use Anthropic (claude) (#2403)

* OPENAI_API_KEY -> ANTHROPIC_API_KEY

* changelog

* ai: rename api key -> anthropic-api-key

* ai: docs

* this, no? (#2473)

---------

Co-authored-by: Taylor Downs <[email protected]>

CHANGELOG.md

@@ -19,6 +19,9 @@ and this project adheres to
 
 ### Changed
 
+- Added Claude integration in job chat
+- OPENAI_API_KEY renamed to AI_ASSISTANT_API_KEY
+
 ### Fixed
 
 ## [v2.10.8] - 2024-12-18

DEPLOYMENT.md

@@ -210,7 +210,8 @@ the Apollo AI service.
 
 The following environment variables are required:
 
-- `OPENAI_API_KEY` - your OpenAI API key.
+- `AI_ASSISTANT_API_KEY` - API key to use the assistant. This currently requires
+  an Anthropic key.
 - `APOLLO_ENDPOINT` - the endpoint for the OpenFn Apollo AI service.
 
 ### Kafka Triggers

lib/lightning/ai_assistant/ai_assistant.ex

@@ -191,7 +191,7 @@ defmodule Lightning.AiAssistant do
   @spec enabled?() :: boolean()
   def enabled? do
     endpoint = Lightning.Config.apollo(:endpoint)
-    api_key = Lightning.Config.apollo(:openai_api_key)
+    api_key = Lightning.Config.apollo(:ai_assistant_api_key)
 
     is_binary(endpoint) && is_binary(api_key)
   end

lib/lightning/apollo_client.ex

@@ -12,7 +12,7 @@ defmodule Lightning.ApolloClient do
   @spec query(String.t(), context(), list()) :: Tesla.Env.result()
   def query(content, context \\ %{}, history \\ []) do
     payload = %{
-      "api_key" => Lightning.Config.apollo(:openai_api_key),
+      "api_key" => Lightning.Config.apollo(:ai_assistant_api_key),
       "content" => content,
       "context" => context,
       "history" => history

lib/lightning/config/bootstrap.ex

@@ -106,7 +106,7 @@ defmodule Lightning.Config.Bootstrap do
           :string,
           Utils.get_env([:lightning, :apollo, :endpoint])
         ),
-      openai_api_key: env!("OPENAI_API_KEY", :string, nil)
+      ai_assistant_api_key: env!("AI_ASSISTANT_API_KEY", :string, nil)
 
     config :lightning, Lightning.Runtime.RuntimeManager,
       start:

test/lightning/ai_assistant/ai_assistant_test.exs

@@ -28,7 +28,7 @@ defmodule Lightning.AiAssistantTest do
       Mox.stub(Lightning.MockConfig, :apollo, fn key ->
         case key do
           :endpoint -> "http://localhost:3000"
-          :openai_api_key -> "api_key"
+          :ai_assistant_api_key -> "api_key"
         end
       end)
 
@@ -73,7 +73,7 @@ defmodule Lightning.AiAssistantTest do
       Mox.stub(Lightning.MockConfig, :apollo, fn key ->
         case key do
           :endpoint -> "http://localhost:3000"
-          :openai_api_key -> "api_key"
+          :ai_assistant_api_key -> "api_key"
         end
       end)
 

test/lightning/apollo_client_test.exs

@@ -11,7 +11,7 @@ defmodule Lightning.ApolloClientTest do
     Mox.stub(Lightning.MockConfig, :apollo, fn key ->
       case key do
         :endpoint -> "http://localhost:3000"
-        :openai_api_key -> "api_key"
+        :ai_assistant_api_key -> "api_key"
       end
     end)
 

test/lightning_web/live/workflow_live/edit_test.exs

@@ -1793,7 +1793,7 @@ defmodule LightningWeb.WorkflowLive.EditTest do
       # when not configured properly
       Mox.stub(Lightning.MockConfig, :apollo, fn
         :endpoint -> nil
-        :openai_api_key -> "openai_api_key"
+        :ai_assistant_api_key -> "ai_assistant_api_key"
       end)
 
       {:ok, view, _html} =
@@ -1811,7 +1811,7 @@ defmodule LightningWeb.WorkflowLive.EditTest do
       # when configured properly
       Mox.stub(Lightning.MockConfig, :apollo, fn
         :endpoint -> "http://localhost:4001"
-        :openai_api_key -> "openai_api_key"
+        :ai_assistant_api_key -> "ai_assistant_api_key"
       end)
 
       {:ok, view, _html} =
@@ -1836,7 +1836,7 @@ defmodule LightningWeb.WorkflowLive.EditTest do
     } do
       Mox.stub(Lightning.MockConfig, :apollo, fn
         :endpoint -> "http://localhost:4001"
-        :openai_api_key -> "openai_api_key"
+        :ai_assistant_api_key -> "ai_assistant_api_key"
       end)
 
       # when disclaimer hasn't been read and no session exists
@@ -1933,7 +1933,7 @@ defmodule LightningWeb.WorkflowLive.EditTest do
 
       Mox.stub(Lightning.MockConfig, :apollo, fn
         :endpoint -> apollo_endpoint
-        :openai_api_key -> "openai_api_key"
+        :ai_assistant_api_key -> "ai_assistant_api_key"
       end)
 
       Mox.stub(
@@ -2160,7 +2160,7 @@ defmodule LightningWeb.WorkflowLive.EditTest do
 
       Mox.stub(Lightning.MockConfig, :apollo, fn
         :endpoint -> apollo_endpoint
-        :openai_api_key -> "openai_api_key"
+        :ai_assistant_api_key -> "ai_assistant_api_key"
       end)
 
       Mox.stub(
@@ -2233,7 +2233,7 @@ defmodule LightningWeb.WorkflowLive.EditTest do
 
       Mox.stub(Lightning.MockConfig, :apollo, fn
         :endpoint -> apollo_endpoint
-        :openai_api_key -> "openai_api_key"
+        :ai_assistant_api_key -> "ai_assistant_api_key"
       end)
 
       expected_question = "Can you help me with this?"
@@ -2314,7 +2314,7 @@ defmodule LightningWeb.WorkflowLive.EditTest do
 
       Mox.stub(Lightning.MockConfig, :apollo, fn
         :endpoint -> apollo_endpoint
-        :openai_api_key -> "openai_api_key"
+        :ai_assistant_api_key -> "ai_assistant_api_key"
       end)
 
       Mox.stub(
@@ -2366,7 +2366,7 @@ defmodule LightningWeb.WorkflowLive.EditTest do
 
       Mox.stub(Lightning.MockConfig, :apollo, fn
         :endpoint -> apollo_endpoint
-        :openai_api_key -> "openai_api_key"
+        :ai_assistant_api_key -> "ai_assistant_api_key"
       end)
 
       Mox.stub(
@@ -2416,7 +2416,7 @@ defmodule LightningWeb.WorkflowLive.EditTest do
     } do
       Mox.stub(Lightning.MockConfig, :apollo, fn
         :endpoint -> "http://localhost:4001/health_check"
-        :openai_api_key -> "openai_api_key"
+        :ai_assistant_api_key -> "ai_assistant_api_key"
       end)
 
       error_message = "You have reached your quota of AI queries"