Skip to content

Commit

Permalink
rename module (getzep#30)
Browse files Browse the repository at this point in the history
Co-authored-by: Daniel Chalef <[email protected]>
  • Loading branch information
danielchalef and Daniel Chalef authored May 12, 2023
1 parent 1bbbd9f commit 3f14209
Show file tree
Hide file tree
Showing 26 changed files with 57 additions and 123 deletions.
2 changes: 1 addition & 1 deletion cmd/zep/cmd.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ package cmd
import (
"os"

"github.com/danielchalef/zep/internal"
"github.com/getzep/zep/internal"
"github.com/sirupsen/logrus"

"github.com/spf13/cobra"
Expand Down
12 changes: 6 additions & 6 deletions cmd/zep/run.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,12 @@ import (
"os/signal"
"syscall"

"github.com/danielchalef/zep/config"
"github.com/danielchalef/zep/pkg/extractors"
"github.com/danielchalef/zep/pkg/llms"
"github.com/danielchalef/zep/pkg/memorystore"
"github.com/danielchalef/zep/pkg/models"
"github.com/danielchalef/zep/pkg/server"
"github.com/getzep/zep/config"
"github.com/getzep/zep/pkg/extractors"
"github.com/getzep/zep/pkg/llms"
"github.com/getzep/zep/pkg/memorystore"
"github.com/getzep/zep/pkg/models"
"github.com/getzep/zep/pkg/server"
)

const (
Expand Down
2 changes: 1 addition & 1 deletion config/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ package config
import (
"strings"

"github.com/danielchalef/zep/internal"
"github.com/getzep/zep/internal"

"github.com/joho/godotenv"
"github.com/sirupsen/logrus"
Expand Down
11 changes: 2 additions & 9 deletions go.mod
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
module github.com/danielchalef/zep
module github.com/getzep/zep

go 1.20

Expand All @@ -8,20 +8,18 @@ require (
github.com/google/uuid v1.3.0
github.com/jinzhu/copier v0.3.5
github.com/joho/godotenv v1.5.1
github.com/oiime/logrusbun v0.1.1
github.com/pgvector/pgvector-go v0.1.1
github.com/pkoukk/tiktoken-go v0.1.1
github.com/sashabaranov/go-openai v1.9.1
github.com/sirupsen/logrus v1.9.0
github.com/spf13/cobra v1.7.0
github.com/spf13/viper v1.15.0
github.com/stretchr/testify v1.8.2
github.com/swaggo/http-swagger v1.3.4
github.com/swaggo/http-swagger/example/go-chi v0.0.0-20230327134356-bc837951e6c7
github.com/swaggo/swag v1.16.1
github.com/uptrace/bun v1.1.12
github.com/uptrace/bun/dialect/pgdialect v1.1.12
github.com/uptrace/bun/driver/pgdriver v1.1.12
google.golang.org/appengine v1.6.7
)

require (
Expand All @@ -33,32 +31,27 @@ require (
github.com/go-openapi/jsonreference v0.20.2 // indirect
github.com/go-openapi/spec v0.20.9 // indirect
github.com/go-openapi/swag v0.22.3 // indirect
github.com/golang/protobuf v1.5.2 // indirect
github.com/hashicorp/hcl v1.0.0 // indirect
github.com/inconshreveable/mousetrap v1.1.0 // indirect
github.com/jinzhu/inflection v1.0.0 // indirect
github.com/josharian/intern v1.0.0 // indirect
github.com/magiconair/properties v1.8.7 // indirect
github.com/mailru/easyjson v0.7.7 // indirect
github.com/mitchellh/mapstructure v1.5.0 // indirect
github.com/oiime/logrusbun v0.1.1 // indirect
github.com/pelletier/go-toml/v2 v2.0.7 // indirect
github.com/pmezard/go-difflib v1.0.0 // indirect
github.com/spf13/afero v1.9.5 // indirect
github.com/spf13/cast v1.5.0 // indirect
github.com/spf13/jwalterweatherman v1.1.0 // indirect
github.com/spf13/pflag v1.0.5 // indirect
github.com/subosito/gotenv v1.4.2 // indirect
github.com/swaggo/files v0.0.0-20220610200504-28940afbdbfe // indirect
github.com/tmthrgd/go-hex v0.0.0-20190904060850-447a3041c3bc // indirect
github.com/vmihailenco/msgpack/v5 v5.3.5 // indirect
github.com/vmihailenco/tagparser/v2 v2.0.0 // indirect
golang.org/x/crypto v0.6.0 // indirect
golang.org/x/net v0.9.0 // indirect
golang.org/x/sys v0.8.0 // indirect
golang.org/x/text v0.9.0 // indirect
golang.org/x/tools v0.8.0 // indirect
google.golang.org/protobuf v1.28.1 // indirect
gopkg.in/ini.v1 v1.67.0 // indirect
gopkg.in/yaml.v3 v3.0.1 // indirect
mellium.im/sasl v0.3.1 // indirect
Expand Down
61 changes: 0 additions & 61 deletions go.sum

Large diffs are not rendered by default.

4 changes: 2 additions & 2 deletions main.go
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
package main

import (
cmd "github.com/danielchalef/zep/cmd/zep"
"github.com/danielchalef/zep/internal"
cmd "github.com/getzep/zep/cmd/zep"
"github.com/getzep/zep/internal"
)

var log = internal.GetLogger()
Expand Down
4 changes: 2 additions & 2 deletions pkg/extractors/embedder.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@ import (
"fmt"
"strings"

"github.com/danielchalef/zep/pkg/llms"
"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/pkg/llms"
"github.com/getzep/zep/pkg/models"
)

// Force compiler to validate that Extractor implements the MemoryStore interface.
Expand Down
8 changes: 4 additions & 4 deletions pkg/extractors/embedder_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,10 @@ import (
"math"
"testing"

"github.com/danielchalef/zep/pkg/llms"
"github.com/danielchalef/zep/pkg/memorystore"
"github.com/danielchalef/zep/pkg/models"
"github.com/danielchalef/zep/test"
"github.com/getzep/zep/pkg/llms"
"github.com/getzep/zep/pkg/memorystore"
"github.com/getzep/zep/pkg/models"
"github.com/getzep/zep/test"
"github.com/spf13/viper"
"github.com/stretchr/testify/assert"
)
Expand Down
2 changes: 1 addition & 1 deletion pkg/extractors/extractor_base.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ import (
"fmt"
"sync"

"github.com/danielchalef/zep/internal"
"github.com/getzep/zep/internal"
)

var log = internal.GetLogger()
Expand Down
2 changes: 1 addition & 1 deletion pkg/extractors/initialize.go
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
package extractors

import (
"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/pkg/models"
)

func Initialize(appState *models.AppState) {
Expand Down
6 changes: 3 additions & 3 deletions pkg/extractors/summarizer.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,9 @@ import (
"fmt"
"strings"

"github.com/danielchalef/zep/internal"
"github.com/danielchalef/zep/pkg/llms"
"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/internal"
"github.com/getzep/zep/pkg/llms"
"github.com/getzep/zep/pkg/models"
)

const SummaryMaxOutputTokens = 512
Expand Down
11 changes: 6 additions & 5 deletions pkg/extractors/summarizer_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,15 @@ package extractors

import (
"context"
"github.com/danielchalef/zep/pkg/llms"
"github.com/danielchalef/zep/pkg/memorystore"
"github.com/danielchalef/zep/pkg/models"
"github.com/danielchalef/zep/test"
"testing"

"github.com/getzep/zep/pkg/llms"
"github.com/getzep/zep/pkg/memorystore"
"github.com/getzep/zep/pkg/models"
"github.com/getzep/zep/test"
"github.com/google/uuid"
"github.com/jinzhu/copier"
"github.com/stretchr/testify/assert"
"testing"
)

func TestSummarize(t *testing.T) {
Expand Down
4 changes: 2 additions & 2 deletions pkg/extractors/token_counter.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@ import (
"context"
"fmt"

"github.com/danielchalef/zep/pkg/llms"
"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/pkg/llms"
"github.com/getzep/zep/pkg/models"
)

// Force compiler to validate that our Extractor implements the Extractor interface.
Expand Down
8 changes: 4 additions & 4 deletions pkg/extractors/token_counter_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,11 @@ import (
"context"
"testing"

"github.com/danielchalef/zep/internal"
"github.com/getzep/zep/internal"

"github.com/danielchalef/zep/pkg/memorystore"
"github.com/danielchalef/zep/pkg/models"
"github.com/danielchalef/zep/test"
"github.com/getzep/zep/pkg/memorystore"
"github.com/getzep/zep/pkg/models"
"github.com/getzep/zep/test"
"github.com/stretchr/testify/assert"
)

Expand Down
2 changes: 1 addition & 1 deletion pkg/llms/embeddings_openai.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ import (
"context"
"fmt"

"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/pkg/models"
"github.com/sashabaranov/go-openai"
)

Expand Down
5 changes: 3 additions & 2 deletions pkg/llms/embeddings_openai_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,12 @@ package llms

import (
"context"
"github.com/danielchalef/zep/test"
"testing"
"time"

"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/test"

"github.com/getzep/zep/pkg/models"
"github.com/stretchr/testify/assert"
)

Expand Down
2 changes: 1 addition & 1 deletion pkg/llms/llm_base.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ package llms
import (
"fmt"

"github.com/danielchalef/zep/internal"
"github.com/getzep/zep/internal"
)

const DefaultTemperature = 0.0
Expand Down
4 changes: 2 additions & 2 deletions pkg/llms/llm_openai.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@ package llms
import (
"context"

"github.com/danielchalef/zep/config"
"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/config"
"github.com/getzep/zep/pkg/models"
"github.com/pkoukk/tiktoken-go"
"github.com/sashabaranov/go-openai"
)
Expand Down
4 changes: 2 additions & 2 deletions pkg/memorystore/base.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@ import (
"context"
"fmt"

"github.com/danielchalef/zep/internal"
"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/internal"
"github.com/getzep/zep/pkg/models"
)

var log = internal.GetLogger()
Expand Down
4 changes: 2 additions & 2 deletions pkg/memorystore/postgres.go
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@ import (

"github.com/sirupsen/logrus"

"github.com/danielchalef/zep/pkg/llms"
"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/pkg/llms"
"github.com/getzep/zep/pkg/models"
"github.com/google/uuid"
"github.com/jinzhu/copier"
"github.com/pgvector/pgvector-go"
Expand Down
10 changes: 5 additions & 5 deletions pkg/memorystore/postgres_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,14 +7,14 @@ import (
"testing"
"time"

"github.com/danielchalef/zep/pkg/extractors"
"github.com/danielchalef/zep/pkg/llms"
"github.com/getzep/zep/pkg/extractors"
"github.com/getzep/zep/pkg/llms"

"github.com/danielchalef/zep/internal"
"github.com/getzep/zep/internal"
"github.com/sirupsen/logrus"

"github.com/danielchalef/zep/pkg/models"
"github.com/danielchalef/zep/test"
"github.com/getzep/zep/pkg/models"
"github.com/getzep/zep/test"
"github.com/google/uuid"
"github.com/spf13/viper"
"github.com/stretchr/testify/assert"
Expand Down
2 changes: 1 addition & 1 deletion pkg/models/appstate.go
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
package models

import (
"github.com/danielchalef/zep/config"
"github.com/getzep/zep/config"
"github.com/sashabaranov/go-openai"
)

Expand Down
4 changes: 2 additions & 2 deletions pkg/server/handlers.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@ import (

"github.com/go-chi/chi/v5"

"github.com/danielchalef/zep/internal"
"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/internal"
"github.com/getzep/zep/pkg/models"
)

var log = internal.GetLogger()
Expand Down
2 changes: 1 addition & 1 deletion pkg/server/routes.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ import (
"time"

httpLogger "github.com/chi-middleware/logrus-logger"
"github.com/danielchalef/zep/pkg/models"
"github.com/getzep/zep/pkg/models"
"github.com/go-chi/chi/v5"
"github.com/go-chi/chi/v5/middleware"
"github.com/spf13/viper"
Expand Down
2 changes: 1 addition & 1 deletion test/data.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
package test

import "github.com/danielchalef/zep/pkg/models"
import "github.com/getzep/zep/pkg/models"

var TestMessages = []models.Message{
{
Expand Down
2 changes: 1 addition & 1 deletion test/utils.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ import (

"github.com/joho/godotenv"

"github.com/danielchalef/zep/config"
"github.com/getzep/zep/config"
)

const TestDsn string = "postgres://postgres:postgres@localhost:5432/postgres?sslmode=disable"
Expand Down

0 comments on commit 3f14209

Please sign in to comment.