Skip to content

Commit

Permalink
New models - gemini 2.0 flash and deepseek v3
Browse files Browse the repository at this point in the history
  • Loading branch information
peterbanda committed Jan 3, 2025
1 parent 66500d8 commit 1249852
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,7 @@ import io.cequence.openaiscala.anthropic.domain.response.{
CreateMessageResponse
}
import io.cequence.openaiscala.anthropic.domain.settings.AnthropicCreateMessageSettings
import io.cequence.openaiscala.anthropic.domain.{ChatRole, Content, Message}
import io.cequence.wsclient.JsonUtil.JsonOps
import io.cequence.openaiscala.anthropic.domain.Message
import io.cequence.wsclient.ResponseImplicits.JsonSafeOps

import scala.concurrent.Future
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ private[service] class OpenAIVertexAIChatCompletionService(
created = openAIResponse.created,
model = openAIResponse.model,
system_fingerprint = openAIResponse.system_fingerprint,
choices = openAIResponse.choices.map(info =>
choices = openAIResponse.choices.map { info =>
ChatCompletionChoiceChunkInfo(
delta = ChunkMessageSpec(
Some(ChatRole.Assistant),
Expand All @@ -73,7 +73,7 @@ private[service] class OpenAIVertexAIChatCompletionService(
index = info.index,
finish_reason = info.finish_reason
)
),
},
usage = openAIResponse.usage
)
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -156,6 +156,7 @@ object NonOpenAIModelId {
val qwen2_72b_instruct = "Qwen/Qwen2-72B-Instruct" // Together AI

// Google Vertex AI
val gemini_2_0_flash_thinking_exp_1219 = "gemini-2.0-flash-thinking-exp-1219"
val gemini_2_0_flash_exp = "gemini-2.0-flash-exp"
val gemini_flash_experimental = "gemini-flash-experimental"
val gemini_pro_experimental = "gemini-pro-experimental"
Expand Down Expand Up @@ -224,6 +225,7 @@ object NonOpenAIModelId {

// Deepseek
// context 64K, 4K (8KBeta)
val deepseek_v3 = "deepseek-v3" // Fireworks
val deepseek_chat = "deepseek-chat"
val deepseek_coder = "deepseek-coder"
}

0 comments on commit 1249852

Please sign in to comment.