Skip to content
This repository has been archived by the owner on Nov 13, 2024. It is now read-only.

Add support OctoAI LLM and embeddings #301

Merged
merged 16 commits into from
Mar 26, 2024
Merged

Conversation

ptorru
Copy link
Contributor

@ptorru ptorru commented Feb 17, 2024

Problem

Currently there is no way to invoke OctoAI endpoints

Solution

Add OctoAI adaptors based on those from Anyscale. Updated README for env vars.

Type of Change

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • This change requires a documentation update
  • Infrastructure change (CI configs, etc)
  • Non-code change (docs, etc)
  • None of the above: (explain here)

Test Plan

Added unit-test for record-encoder. Can be checked via:

poetry run pytest tests/system/record_encoder/test_octoai_record_encoder.py

Done manual e-2-e test using canopy server/chat.

Use the following .env

PINECONE_API_KEY="..."
OCTOAI_API_KEY="..."
INDEX_NAME="octo"
CANOPY_CONFIG_FILE="src/canopy/config_templates/octoai.yaml"
OPENAI_API_KEY="..."

Then:

poetry run canopy new
poetry run canopy upsert doc.txt
poetry run canopy start

In a separate terminal:

poetry run canopy chat --no-stream

Copy link
Collaborator

@izellevy izellevy left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sorry for the late response. Please see some suggestions

src/canopy/knowledge_base/record_encoder/octoai.py Outdated Show resolved Hide resolved
src/canopy/llm/octoai.py Outdated Show resolved Hide resolved
src/canopy/llm/octoai.py Show resolved Hide resolved
@izellevy izellevy self-requested a review March 5, 2024 09:28
@izellevy izellevy added this pull request to the merge queue Mar 5, 2024
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Mar 5, 2024
@izellevy izellevy added this pull request to the merge queue Mar 12, 2024
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Mar 12, 2024
@izellevy izellevy added this pull request to the merge queue Mar 26, 2024
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Mar 26, 2024
@izellevy izellevy added this pull request to the merge queue Mar 26, 2024
Merged via the queue into pinecone-io:main with commit 73338bb Mar 26, 2024
7 checks passed
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants