-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
New AI Chat with Anthropic Claude block
- Loading branch information
1 parent
053687a
commit 947e815
Showing
3 changed files
with
164 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,12 @@ | ||
| ### AI Chat with Anthropic Claude | ||
|
|
||
| #### Executes a one-off prompt request for each row in the input block, and outputs the results using a Anthropic Claude model | ||
|
|
||
| ## Language | ||
| Python | ||
|
|
||
| ## Dependencies | ||
| openai | ||
|
|
||
| ## Source | ||
| [script.py](https://github.com/visokio/omniscope-custom-blocks/blob/master/Connectors/AI%20Chat%0Anthropic%20Claude/script.py) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,102 @@ | ||
| { | ||
| "@visokiotype": "CustomBlockSchema.CustomBlockManifest", | ||
| "name": "AI Chat - Claude", | ||
| "scriptFilename": "script.py", | ||
| "language": "PYTHON", | ||
| "executableVersion": null, | ||
| "minVersions": [ | ||
| null | ||
| ], | ||
| "optionsVersion": 1, | ||
| "apiVersion": "VERSION_0", | ||
| "isResourceIntensiveScript": false, | ||
| "showPartitioning": false, | ||
| "icon": "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAKEAAAB7CAYAAAAc9OD/AAAABmJLR0QA/wD/AP+gvaeTAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH6AMWDDMFiQopBwAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAFxUlEQVR42u2d4XLcIAyEQfimff+3bXKG/nAzyWTSNnesYAWrB7jD4rOkxYByay3JZDPN5AKZIJQJQrlAJghlglAukAlCmSCUC2SCUCYI5QKZIJQJQrlAJghlglAukM22Y4WHuN9fWmsV75zjR845d/1Grfd2nnfouG63n3klCJeIhB4AvgHU7WA7MuO4BGGQCan1hPxOzlg3oyOrICSfkFrPbshLwVc9rdUmCDnSsPtE1NoPec6WU8qhXj5BSDQR1/GHftZLKSHqYEFIOhHnKYEiCCdPAK9AOZMgnJqKz8HQMwqUtoRACQnh5fg2GkKQQOGM0oIwgOOv+rMffLNDEArCrhIAkJKLBEp0CGc6HLFmmFKWQIkO4WyHS6BsDuEMQSKBIgjpHC2BIghZSgJSgXI2QUhei60uUDDjEoT/iD5cDkYIATP0poaWIt43GQJCBkHi8VKYac0wDISMRTdqB48EiiDsjYYSKDtAyOxQCZRtIOR2qATK4hB6OBM94RIoi0OIdqRZSaVgt9njBEoB+y6OQCGHEOvIa6Jz6r1VweNl8TmDEkOgGDGAcAe+bRxAL4sgdvbk7PFyxIiGxBBiBclH8PA1WAMJFOzLgdpssSWEHoLk83rcLgIFsZa5JYRoQXKtw2VXISCBshyEHoLkq/pQAkUQDnLY39Ic+moOCZR1IHSPgn5RRwJlAQgb/H6Z/4GG/257Il4cB4HCm5KpIEQrue+kNvTJN1QkxwsU3m/wVBDiU/HxDVCNUgigIWQWKEYE4DBB4l2DIdYMfdQ7p0BhgnCYIPlCJVMKFHSpwCpQSCD0ECSPpLMsgbI7hHjH5IdrPdYvFTsIFAoI0Y55ZhHaZ2Mpp0Bhu7dmOoQeDnl2EZox6uxw8/90CNEO6antWM96rH7z/2QI8YKkR1HutKmBaYvXVAg9lFrv4jM66kigkEOI3z1dAL9xSKDsAqGHA1An6RgPo68sUKZBiBckOaEmCf+lQgKFNRKCU/EBBNooD6Ov2ppsCoQeD45ebGa9LWvF3slTIEQ/uEfRznpb1oq9k4dD6POFpDiMlHNTg0epMDsaDocQ/8DZZWI84EZtpfLZ4rVXJESnTT/nkB5GX02gDIXQR5Ac2dVBpFu8VmpNNhRC9IOiJ2Jc1FFrso92jPojjxv4W2vpfn8Jd1t9rWd3hP1TBzf0uEqxdSH0OWQTs2/Hu0DpC7JmB/T7+wXhbd10vEJzaHBpot7JIyGM3hTaySeIqngJgWKD3npRJ4EyD0LGlmBMAgWwQhC+d7JFcPSqpt7JglACxblUoIAwahNoCZSxmxrM9y2/i7Jv1829KRn9DX2cQDFfxyoQjnpZnW6QiB0JVQs+KlASAkRBKAi7akPEmmFIgWJOACoPT6mfY/ZOdoJQgmQVgTKid7LhHRlzZ4sEim+pMBRCbVaQQHnUDjyE+AGP2EE9G5yPL3Hvbm6zI6Pr8lrP5hFl4RB6CJJSbsnr4RH2+voLuh56nmf3t+C3/i3IsqjWu9PRWnA69hAkzABeLwnnlwp8azK/Wt+wzmtoAOnrOKeDUFsJFBiEHjd/eh/nZK1ZeS/W9BEoxjpAj5arfikZf15sp97JRjswO1IU87kfZp/eyUY7MHJB4l+/7tM72UDO2k6QfJGStxEo6AvvjW1AUSG8Ng9wpj72m/+NbUBeNdaYlOwhUNbvnWwsA4koSMakvvV7JxvLQN5rq5JTYGMVKD43/7fZEOIFybXoG5pBr9SHyDC0AsVmDyC+IImT+lgFis0egHdNNcM8rjBeuTWZzfrjFaOgZ+pbuTWZzfrjERM3OS2DI866rcls1h9/mrIwmxXmCpQ1eyfb6D8cVUMRQEh7IwLbsdAnVoYb/CFWS8XvL9eRasVmjdZq628sfsBLqtba09ks63imbHrWkAtkglAmCOUCmSCUCUK5QCYIZYJQLpAJQpkglAtkglAmCOUCmSCUCUK5QDbbfgPuDjK/9b41hwAAAABJRU5ErkJggg==", | ||
| "description": "Executes a one-off prompt to Anthropic Claude and returns the generated text result", | ||
| "category": "Code & AI", | ||
| "subcategory": null, | ||
| "tags": [ | ||
| "LLM", | ||
| "NLP", | ||
| "Anthropic", | ||
| "Claude" | ||
| ], | ||
| "introductoryText": "\n### AI Chat with Anthropic Claude\n#### Executes a one-off prompt request for each row in the input block, and output the results from Claude using the specified text generation / chat completion model", | ||
| "dependencies": "anthropic", | ||
| "options": [ | ||
| { | ||
| "name": "AnthropicApiKey", | ||
| "title": "Anthropic API Key", | ||
| "description": "Find it here https://console.anthropic.com/settings/keys", | ||
| "groupTitle": null, | ||
| "width": "ONE_AND_HALF", | ||
| "@visokiotype": "CustomBlockSchema.PasswordCustomBlockPublicOption", | ||
| "mandatory": true | ||
| }, | ||
| { | ||
| "name": "Model", | ||
| "title": "Model", | ||
| "description": "E.g. claude-3-opus-20240229 . The model as per Anthropic API name.", | ||
| "groupTitle": null, | ||
| "width": "ONE", | ||
| "@visokiotype": "CustomBlockSchema.TextCustomBlockPublicOption", | ||
| "mandatory": true, | ||
| "defaultValue": "claude-3-opus-20240229" | ||
| }, | ||
| { | ||
| "name": "Temperature", | ||
| "title": "Temperature", | ||
| "description": "The model temperature typically from 0 to 1", | ||
| "groupTitle": null, | ||
| "width": "ONE", | ||
| "@visokiotype": "CustomBlockSchema.DoubleCustomBlockPublicOption", | ||
| "numMin": 0, | ||
| "numMax": 1, | ||
| "mandatory": true, | ||
| "defaultValue": 0.5, | ||
| "showAsSlider": true | ||
| }, | ||
| { | ||
| "name": "SystemPrompt", | ||
| "title": "System Prompt", | ||
| "description": "Helps set the behavior of the model / AI assistant", | ||
| "groupTitle": null, | ||
| "width": "ONE", | ||
| "@visokiotype": "CustomBlockSchema.FieldCustomBlockPublicOption", | ||
| "mandatory": false, | ||
| "inputIndex": 0, | ||
| "defaultValue": null, | ||
| "fieldTypes": [] | ||
| }, | ||
| { | ||
| "name": "Prompt", | ||
| "title": "User Prompt", | ||
| "description": "Point to the field that contains the prompt to execute", | ||
| "groupTitle": null, | ||
| "width": "ONE_AND_HALF", | ||
| "@visokiotype": "CustomBlockSchema.FieldCustomBlockPublicOption", | ||
| "mandatory": true, | ||
| "inputIndex": 0, | ||
| "defaultValue": null, | ||
| "fieldTypes": [] | ||
| } | ||
| ], | ||
| "blockOutputs": [ | ||
| { | ||
| "@visokiotype": "CustomBlockSchema.BlockOutputPublicOption", | ||
| "id": "Output Data", | ||
| "label": "1", | ||
| "displayName": "Output", | ||
| "tooltip": null | ||
| } | ||
| ], | ||
| "docker": { | ||
| "@visokiotype": "CustomBlockSchema.DockerCustomBlockPublicOption", | ||
| "customBaseImage": null, | ||
| "useCustomBaseImage": false, | ||
| "customSystemLibraries": null, | ||
| "installVisokioRepLibraries": false | ||
| }, | ||
| "designLock": false, | ||
| "apiMode": "BATCH" | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,50 @@ | ||
| import pandas as pd | ||
| from omniscope.api import OmniscopeApi | ||
| omniscope_api = OmniscopeApi() | ||
|
|
||
| # read the records associated to the first block input | ||
| input_data = omniscope_api.read_input_records(input_number=0) | ||
|
|
||
| API_Key = omniscope_api.get_option("AnthropicApiKey") | ||
| prompt = omniscope_api.get_option("Prompt") | ||
| systemPrompt = omniscope_api.get_option("SystemPrompt") | ||
| model = omniscope_api.get_option("Model") | ||
| temperature = omniscope_api.get_option("Temperature") | ||
|
|
||
| import anthropic | ||
|
|
||
| client = anthropic.Anthropic( | ||
| api_key=API_Key, | ||
| ) | ||
|
|
||
| result_df = [] | ||
| for index, row in input_data.iterrows(): | ||
| the_prompt = row[prompt] | ||
| the_systemprompt = row[systemPrompt] | ||
|
|
||
| message = client.messages.create( | ||
| model=model, | ||
| max_tokens=1000, | ||
| temperature=temperature, | ||
| system=the_systemprompt, | ||
| messages=[ | ||
| { | ||
| "role": "user", | ||
| "content": [ | ||
| { | ||
| "type": "text", | ||
| "text": the_prompt | ||
| } | ||
| ] | ||
| } | ||
| ] | ||
| ) | ||
| result_df.append({"Prompt" : the_prompt, "System Prompt": the_systemprompt, "Response" : message.content[0].text}) | ||
|
|
||
| output_data = pd.DataFrame(result_df) | ||
|
|
||
|
|
||
| #write the output records in the first output | ||
| if output_data is not None: | ||
| omniscope_api.write_output_records(output_data, output_number=0) | ||
| omniscope_api.close() |