mirror of
https://github.com/sourcegraph/sourcegraph.git
synced 2026-02-06 19:21:50 +00:00
f2590cbb36
CLOSE https://github.com/sourcegraph/cody-issues/issues/211 & https://github.com/sourcegraph/cody-issues/issues/412 & https://github.com/sourcegraph/cody-issues/issues/412 UNBLOCK https://github.com/sourcegraph/cody/pull/4360 * Add support for Google Gemini AI models as chat completions provider * Add new `google` package to handle Google Generative AI client * Update `client.go` and `codygateway.go` to handle the new Google provider * Set default models for chat, fast chat, and completions when Google is the configured provider * Add gemini-pro to the allowed list <!-- 💡 To write a useful PR description, make sure that your description covers: - WHAT this PR is changing: - How was it PREVIOUSLY. - How it will be from NOW on. - WHY this PR is needed. - CONTEXT, i.e. to which initiative, project or RFC it belongs. The structure of the description doesn't matter as much as covering these points, so use your best judgement based on your context. Learn how to write good pull request description: https://www.notion.so/sourcegraph/Write-a-good-pull-request-description-610a7fd3e613496eb76f450db5a49b6e?pvs=4 --> ## Test plan <!-- All pull requests REQUIRE a test plan: https://docs-legacy.sourcegraph.com/dev/background-information/testing_principles --> For Enterprise instances using google as provider: 1. In your Soucegraph local instance's Site Config, add the following: ``` "accessToken": "REDACTED", "chatModel": "gemini-1.5-pro-latest", "provider": "google", ``` Note: You can get the accessToken for Gemini API in 1Password. 2. After saving the site config with the above change, run the following curl command: ``` curl 'https://sourcegraph.test:3443/.api/completions/stream' -i \ -X POST \ -H 'authorization: token $LOCAL_INSTANCE_TOKEN' \ --data-raw '{"messages":[{"speaker":"human","text":"Who are you?"}],"maxTokensToSample":30,"temperature":0,"stopSequences":[],"timeoutMs":5000,"stream":true,"model":"gemini-1.5-pro-latest"}' ``` 3. Expected Output: ``` ❯ curl 'https://sourcegraph.test:3443/.api/completions/stream' -i \ -X POST \ -H 'authorization: token <REDACTED>' \ --data-raw '{"messages":[{"speaker":"human","text":"Who are you?"}],"maxTokensToSample":30,"temperature":0,"stopSequences":[],"timeoutMs":5000,"stream":true,"model":"gemini-1.5-pro-latest"}' HTTP/2 200 access-control-allow-credentials: true access-control-allow-origin: alt-svc: h3=":3443"; ma=2592000 cache-control: no-cache content-type: text/event-stream date: Tue, 04 Jun 2024 05:45:33 GMT server: Caddy server: Caddy vary: Accept-Encoding, Authorization, Cookie, Authorization, X-Requested-With, Cookie x-accel-buffering: no x-content-type-options: nosniff x-frame-options: DENY x-powered-by: Express x-trace: d4b1f02a3e2882a3d52331335d217b03 x-trace-span: 728ec33860d3b5e6 x-trace-url: https://sourcegraph.test:3443/-/debug/jaeger/trace/d4b1f02a3e2882a3d52331335d217b03 x-xss-protection: 1; mode=block event: completion data: {"completion":"I","stopReason":"STOP"} event: completion data: {"completion":"I am a large language model, trained by Google. \n\nThink of me as","stopReason":"STOP"} event: completion data: {"completion":"I am a large language model, trained by Google. \n\nThink of me as a computer program that can understand and generate human-like text.","stopReason":"MAX_TOKENS"} event: done data: {} ``` Verified locally:  #### Before Cody Gateway returns `no client known for upstream provider google` ```sh curl -X 'POST' -d '{"messages":[{"speaker":"human","text":"Who are you?"}],"maxTokensToSample":30,"temperature":0,"stopSequences":[],"timeoutMs":5000,"stream":true,"model":"google/gemini-1.5-pro-latest"}' -H 'Accept: application/json' -H 'Authorization: token $YOUR_DOTCOM_TOKEN' -H 'Content-Type: application/json' 'https://sourcegraph.com/.api/completions/stream' event: error data: {"error":"no client known for upstream provider google"} event: done data: { ``` ## Changelog <!-- 1. Ensure your pull request title is formatted as: $type($domain): $what 2. Add bullet list items for each additional detail you want to cover (see example below) 5. You can edit this after the pull request was merged, as long as release shipping it hasn't been promoted to the public. 6. For more information, please see this how-to https://www.notion.so/sourcegraph/Writing-a-changelog-entry-dd997f411d524caabf0d8d38a24a878c? Audience: TS/CSE > Customers > Teammates (in that order). Cheat sheet: $type = chore|fix|feat $domain: source|search|ci|release|plg|cody|local|... --> <!-- Example: Title: fix(search): parse quotes with the appropriate context Changelog section: ## Changelog - When a quote is used with regexp pattern type, then ... - Refactored underlying code. --> Added support for Google as an LLM provider for Cody, with the following models available through Cody Gateway: Gemini Pro (`gemini-pro-latest`), Gemini 1.5 Flash (`gemini-1.5-flash-latest`), and Gemini 1.5 Pro (`gemini-1.5-pro-latest`). |
||
|---|---|---|
| .. | ||
| accesstoken | ||
| actor | ||
| adminanalytics | ||
| api | ||
| appliance | ||
| audit | ||
| auth | ||
| authbearer | ||
| authz | ||
| batches | ||
| binary | ||
| bytesize | ||
| byteutils | ||
| cloud | ||
| cmd | ||
| codeintel | ||
| codemonitors | ||
| codygateway | ||
| collections | ||
| comby | ||
| completions | ||
| compute | ||
| conf | ||
| cookie | ||
| ctags_config | ||
| database | ||
| debugserver | ||
| deviceid | ||
| diskcache | ||
| diskusage | ||
| dotcom | ||
| download | ||
| embeddings | ||
| encryption | ||
| endpoint | ||
| env | ||
| errcode | ||
| eventlogger | ||
| executor | ||
| extsvc | ||
| featureflag | ||
| fileutil | ||
| github_apps | ||
| gitserver | ||
| goroutine | ||
| gosyntect | ||
| gqltestutil | ||
| gqlutil | ||
| grpc | ||
| guardrails | ||
| hashutil | ||
| highlight | ||
| honey | ||
| hostmatcher | ||
| hostname | ||
| htmlutil | ||
| httpcli | ||
| httpserver | ||
| httptestutil | ||
| insights | ||
| instrumentation | ||
| ipynb | ||
| jsonc | ||
| k8s/resource | ||
| lazyregexp | ||
| license | ||
| licensing | ||
| limiter | ||
| logging | ||
| luasandbox | ||
| mapfs | ||
| maps | ||
| markdown | ||
| memo | ||
| metrics | ||
| notebooks | ||
| oauthtoken | ||
| oauthutil | ||
| observation | ||
| oobmigration | ||
| opencodegraph | ||
| otlpenv | ||
| own | ||
| packagefilters | ||
| paths | ||
| perforce | ||
| productsubscription | ||
| profiler | ||
| pubsub | ||
| randstring | ||
| ratelimit | ||
| rbac | ||
| rcache | ||
| redislock | ||
| redispool | ||
| repos | ||
| repoupdater | ||
| requestclient | ||
| requestinteraction | ||
| rockskip | ||
| sams | ||
| sanitycheck | ||
| search | ||
| searcher | ||
| security | ||
| service | ||
| session | ||
| settings | ||
| siteid | ||
| slack | ||
| slices | ||
| sourcegraphoperator | ||
| src-cli | ||
| src-prometheus | ||
| suspiciousnames | ||
| symbols | ||
| sysreq | ||
| telemetry | ||
| telemetrygateway | ||
| temporarysettings | ||
| testutil | ||
| timeutil | ||
| trace | ||
| tracer | ||
| ttlcache | ||
| txemail | ||
| types | ||
| unpack | ||
| updatecheck | ||
| uploadhandler | ||
| uploadstore | ||
| usagestats | ||
| users | ||
| vcs | ||
| version | ||
| webhooks/outbound | ||
| workerutil | ||
| wrexec | ||
| buf.yaml | ||
| BUILD.bazel | ||