discourse/discourse-ai
1
0
Fork 0
mirror of https://github.com/discourse/discourse-ai.git synced 2025-06-28 18:42:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

FEATURE: Claude 2 for summarization and AIHelper (#101)

Browse Source
This commit is contained in:
Rafael dos Santos Silva 2023-07-13 12:32:08 -03:00 committed by GitHub
parent 1c017c22b0
commit 9d10a152b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/settings.yml
View File

@ -134,7 +134,7 @@ plugins:
choices: choices:
- gpt-3.5-turbo - gpt-3.5-turbo
- gpt-4 - gpt-4
- claude-v1 - claude-2
ai_embeddings_enabled: ai_embeddings_enabled:
default: false default: false

1
lib/modules/summarization/entry_point.rb
View File

@ -22,6 +22,7 @@ module DiscourseAi
Models::Discourse.new("long-t5-tglobal-base-16384-book-summary", max_tokens: 16_384), Models::Discourse.new("long-t5-tglobal-base-16384-book-summary", max_tokens: 16_384),
Models::Anthropic.new("claude-v1", max_tokens: 9000), Models::Anthropic.new("claude-v1", max_tokens: 9000),
Models::Anthropic.new("claude-v1-100k", max_tokens: 100_000), Models::Anthropic.new("claude-v1-100k", max_tokens: 100_000),
Models::Anthropic.new("claude-2", max_tokens: 100_000),
] ]
foldable_models.each do |model| foldable_models.each do |model|

2
lib/shared/inference/anthropic_completions.rb
View File

@ -8,7 +8,7 @@ module ::DiscourseAi
def self.perform!( def self.perform!(
prompt, prompt,
model = "claude-v1", model = "claude-2",
temperature: nil, temperature: nil,
top_p: nil, top_p: nil,
max_tokens: nil, max_tokens: nil,