1 files changed, 5 insertions, 6 deletions
diff --git a/src/client/vertexai_claude.rs b/src/client/vertexai_claude.rs
index bdce7d8..3993078 100644
--- a/src/client/vertexai_claude.rs
+++ b/src/client/vertexai_claude.rs
@@ -1,8 +1,7 @@
-use super::{
-    access_token::*, claude::*, vertexai::*, ChatCompletionsData, ChatCompletionsOutput, Client,
-    ExtraConfig, Model, ModelData, ModelPatches, PromptAction, PromptKind, SseHandler,
-    VertexAIClaudeClient,
-};
+use super::*;
+use super::access_token::*;
+use super::claude::*;
+use super::vertexai::*;
 
 use anyhow::Result;
 use async_trait::async_trait;
@@ -46,7 +45,7 @@ impl VertexAIClaudeClient {
         );
 
         let mut body = claude_build_chat_completions_body(data, &self.model)?;
-        self.patch_request_body(&mut body);
+        self.patch_chat_completions_body(&mut body);
         if let Some(body_obj) = body.as_object_mut() {
             body_obj.remove("model");
         }