Commit Graph
4 Commits
Author SHA1 Message Date
fheinecke b178b8b732 Updated Teleport codebase to AGPL3 license (#35259)
Signed-off-by: Fred Heinecke <fred.heinecke@goteleport.com>
2023-12-01 17:48:14 +00:00
JoelandZac Bergquist 2fd8f75e38 [Assist] Generalize embeddings system (#30835)
* generalize embeddings

* add test for mapping

* use existing mapper

* add streaming apis

* update embedding processor tests

* delete unused code

* fix tests

* Update lib/ai/embeddingprocessor.go

Co-authored-by: Zac Bergquist <zac.bergquist@goteleport.com>

* Update lib/ai/embeddingprocessor.go

Co-authored-by: Zac Bergquist <zac.bergquist@goteleport.com>

* Update lib/ai/embeddingprocessor.go

Co-authored-by: Zac Bergquist <zac.bergquist@goteleport.com>

* fix feedback

---------

Co-authored-by: Zac Bergquist <zac.bergquist@goteleport.com>
2023-08-23 20:01:56 +00:00
Justinas StankevičiusandAlan Parra 18016e52e8 Fix node equality check in embedding processor (#30325)
* Fix node equality check in embedding processor

* Apply minor suggestions from code review

* Make process() private, test using only public API

* Remove an unused function

* Update lib/ai/embeddingprocessor.go

Co-authored-by: Alan Parra <alan.parra@goteleport.com>

---------

Co-authored-by: Alan Parra <alan.parra@goteleport.com>
2023-08-16 10:41:53 +00:00
Hugo Shaka 8dbd3c3069 assist: move embeddings in a separate library, add tool config, opportunistic node lookup (#29716)
* assist: extract embedding types into a dedicated module

* assist: allow the embedding tool to query nodes from proxy cache

This commit does two things:
- Introduce a new toolsConfig structure to pass all required clients to
  the tools. This can also be used to enable or disable specicifc tools,
  which helps with testing
- Pass the proxy node cache and the request access checker to the
  embedding tool so it can try to lookup nodes from cache in if the user
  has less than 10 nodes available.

* Add tests for the opportunistic proxy node lookup

* Apply suggestions from code review

* Update lib/ai/client.go
2023-08-09 18:21:08 +00:00