Mentions, lint, etc.

embeddings/vertexai_palm_test.go

@@ -7,7 +7,7 @@ import (
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	"github.com/tmc/langchaingo/llms/vertexai"
+	"github.com/tmc/langchaingo/llms/googleai/palm"
 )
 
 func newVertexEmbedder(t *testing.T, opts ...Option) *EmbedderImpl {
@@ -17,7 +17,7 @@ func newVertexEmbedder(t *testing.T, opts ...Option) *EmbedderImpl {
 		return nil
 	}
 
-	llm, err := vertexai.New()
+	llm, err := palm.New()
 	require.NoError(t, err)
 
 	embedder, err := NewEmbedder(llm, opts...)

llms/doc.go

@@ -6,7 +6,7 @@
 // 1. Hugging Face:      llms/huggingface/
 // 2. Local LLM:         llms/local/
 // 3. OpenAI:            llms/openai/
-// 4. Vertex AI:         llms/vertexai/
+// 4. Google AI:         llms/googleai/
 // 5. Cohere:            llms/cohere/
 //
 // Each subpackage includes provider-specific LLM implementations and helper files for communication

llms/googleai/palm/palm_llm.go

@@ -1,3 +1,6 @@
+// package palm implements a langchaingo provider for Google Vertex AI legacy
+// PaLM models. Use the newer Gemini models via llms/googleai/vertex if
+// possible.
 package palm
 
 import (

llms/googleai/vertex/new.go

@@ -1,3 +1,6 @@
+// package vertex implements a langchaingo provider for Google Vertex AI LLMs,
+// including the new Gemini models.
+// See https://cloud.google.com/vertex-ai for more details.
 package vertex
 
 import (

llms/googleai/vertex/vertex.go

@@ -1,6 +1,7 @@
-// DO NOT EDIT: this code is auto-generated from llms/googleai/googleai_llm.go
 package vertex
 
+// DO NOT EDIT: this code is auto-generated from llms/googleai/googleai_llm.go
+
 import (
 	"context"
 	"errors"