Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

refactor llm info #961

Merged
merged 23 commits into from
Dec 21, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
6cce2d1
llm as judge support
pelikhan Dec 19, 2024
9dcfc67
Merge remote-tracking branch 'origin/main' into judge
pelikhan Dec 20, 2024
9b572ef
style: πŸ’„ remove redundant blank line in judge.genai.mts
pelikhan Dec 20, 2024
5725842
feat: πŸ§ͺ extend judgeClassify to support WorkspaceFile
pelikhan Dec 20, 2024
e17d55b
feat: πŸ—οΈ add check for undefined text in tracePromptResult
pelikhan Dec 20, 2024
f16d915
add g2thinking
pelikhan Dec 20, 2024
c62171d
feat: ✨ add model alias support and update defaults
pelikhan Dec 20, 2024
0d394b9
feat: ✨ add support for listing model information in CLI
pelikhan Dec 20, 2024
d6da24c
updated list of models
pelikhan Dec 20, 2024
f034b86
feat: ✨ update LLM aliases and models, fix mistral typo
pelikhan Dec 20, 2024
fd17d69
feat: ✨ add logprobs and topLogprobs support
pelikhan Dec 20, 2024
aa7f334
feat: ✨ add aliases and logit_bias support to providers
pelikhan Dec 21, 2024
c7958a1
feat: πŸ”§ add `info models alias` command to CLI
pelikhan Dec 21, 2024
c3a2ca5
test: ♻️ simplify assertion and add new test case
pelikhan Dec 21, 2024
b1c78f0
chore: πŸ”§ update dependencies across multiple packages
pelikhan Dec 21, 2024
500f3fb
feat: ✨ add support for LLM provider aliases and options
pelikhan Dec 21, 2024
c5d16bf
feat: ✨ add embeddings model to llms.json configuration
pelikhan Dec 21, 2024
7c5d691
feat: πŸ“ add filter for empty user text and new aliases
pelikhan Dec 21, 2024
4e890ff
use provider in test scripts
pelikhan Dec 21, 2024
050f7f3
docs: ✏️ update CLI usage and configuration sections
pelikhan Dec 21, 2024
042a338
ci: πŸ”„ rename workflow to huggingface tests
pelikhan Dec 21, 2024
d0a02a9
env typo
pelikhan Dec 21, 2024
2408a79
feat: πŸ”§ update aliases and simplify template tracing
pelikhan Dec 21, 2024
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 1 addition & 5 deletions .github/workflows/anthropic.yml
Original file line number Diff line number Diff line change
Expand Up @@ -32,10 +32,6 @@ jobs:
- name: compile
run: yarn compile
- name: poem
run: yarn run:script poem --model anthropic:claude-3-5-sonnet-20240620 --out-trace $GITHUB_STEP_SUMMARY
env:
ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
- name: cache
run: yarn run:script summarize-cached --model anthropic:claude-3-5-sonnet-20240620 --small-model anthropic:claude-3-5-sonnet-20240620
run: yarn run:script poem -p anthropic --out-trace $GITHUB_STEP_SUMMARY
env:
ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
2 changes: 1 addition & 1 deletion .github/workflows/google.yml
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,6 @@ jobs:
- name: compile
run: yarn compile
- name: poem
run: yarn run:script poem --model google:gemini-1.5-flash --out-trace $GITHUB_STEP_SUMMARY
run: yarn run:script poem -p google --out-trace $GITHUB_STEP_SUMMARY
env:
GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
37 changes: 37 additions & 0 deletions .github/workflows/huggingface.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
name: huggingface tests
on:
workflow_dispatch:
release:
types:
- published
pull_request:
paths:
- yarn.lock
- ".github/workflows/huggingface.yml"
- "packages/core/**/*"
- "packages/cli/**/*"
- "packages/samples/**/*"
concurrency:
group: ${{ github.workflow }}-${{ github.ref }}-huggingface
cancel-in-progress: true
jobs:
tests:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
with:
submodules: "recursive"
fetch-depth: 0
- uses: actions/setup-node@v4
with:
node-version: "20"
cache: yarn
- run: yarn install --frozen-lockfile
- name: typecheck
run: yarn typecheck
- name: compile
run: yarn compile
- name: poem
run: yarn run:script poem -p huggingface --out-trace $GITHUB_STEP_SUMMARY
env:
HUGGINGFACE_API_KEY: ${{ secrets.HUGGINGFACE_API_KEY }}
22 changes: 3 additions & 19 deletions .github/workflows/openai.yml
Original file line number Diff line number Diff line change
@@ -1,24 +1,8 @@
name: openai smoke tests
on:
workflow_dispatch:
# pull_request:
# paths:
# - yarn.lock
# - ".github/workflows/github-models.yml"
# - "packages/core/**/*"
# - "packages/cli/**/*"
# - "packages/samples/**/*"
# push:
# branches:
# - main
# paths:
# - yarn.lock
# - ".github/workflows/github-models.yml"
# - "packages/core/**/*"
# - "packages/cli/**/*"
# - "packages/samples/**/*"
concurrency:
group: ${{ github.workflow }}-${{ github.ref }}-github-models
group: ${{ github.workflow }}-${{ github.ref }}-openai
cancel-in-progress: true
jobs:
tests:
Expand All @@ -37,8 +21,8 @@ jobs:
run: yarn typecheck
- name: compile
run: yarn compile
- name: run summarize openai:gpt-3.5-turbo
run: yarn test:summarize --model openai:gpt-3.5-turbo --out ./temp
- name: run summarize
run: yarn test:summarize -p openai --out-trace $GITHUB_STEP_SUMMARY
env:
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
OPENAI_API_TYPE: ${{ secrets.OPENAI_API_TYPE }}
Expand Down
82 changes: 41 additions & 41 deletions THIRD_PARTY_LICENSES.md
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,7 @@ The following npm packages may be included in this product:

- @aws-crypto/[email protected]
- @aws-crypto/[email protected]
- @huggingface/[email protected].0
- @huggingface/[email protected].1
- [email protected]
- [email protected]

Expand Down Expand Up @@ -1954,25 +1954,25 @@ Apache License

The following npm packages may be included in this product:

- @aws-sdk/client-cognito-identity@3.714.0
- @aws-sdk/client-sso-oidc@3.714.0
- @aws-sdk/client-sso@3.714.0
- @aws-sdk/client-sts@3.714.0
- @aws-sdk/credential-provider-cognito-identity@3.714.0
- @aws-sdk/client-cognito-identity@3.716.0
- @aws-sdk/client-sso-oidc@3.716.0
- @aws-sdk/client-sso@3.716.0
- @aws-sdk/client-sts@3.716.0
- @aws-sdk/credential-provider-cognito-identity@3.716.0
- @aws-sdk/[email protected]
- @aws-sdk/util-user-agent-node@3.714.0
- @aws-sdk/util-user-agent-node@3.716.0
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected].30
- @smithy/[email protected].31
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected].30
- @smithy/[email protected].30
- @smithy/[email protected].31
- @smithy/[email protected].31

These packages each contain the following license:

Expand Down Expand Up @@ -2182,7 +2182,7 @@ Apache License

The following npm package may be included in this product:

- @aws-sdk/client-bedrock-runtime@3.714.0
- @aws-sdk/client-bedrock-runtime@3.716.0

This package contains the following license:

Expand Down Expand Up @@ -2394,7 +2394,7 @@ The following npm packages may be included in this product:

- @aws-sdk/[email protected]
- @aws-sdk/[email protected]
- @aws-sdk/middleware-user-agent@3.714.0
- @aws-sdk/middleware-user-agent@3.716.0
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected]
Expand All @@ -2408,7 +2408,7 @@ The following npm packages may be included in this product:
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected].0
- @smithy/[email protected].1
- @smithy/[email protected]
- @smithy/[email protected]

Expand Down Expand Up @@ -2861,7 +2861,7 @@ MIT License

The following npm package may be included in this product:

- [email protected].0
- [email protected].4

This package contains the following license:

Expand Down Expand Up @@ -4161,10 +4161,10 @@ Apache License

The following npm packages may be included in this product:

- @aws-sdk/credential-provider-env@3.714.0
- @aws-sdk/credential-provider-ini@3.714.0
- @aws-sdk/credential-provider-node@3.714.0
- @aws-sdk/credential-providers@3.715.0
- @aws-sdk/credential-provider-env@3.716.0
- @aws-sdk/credential-provider-ini@3.716.0
- @aws-sdk/credential-provider-node@3.716.0
- @aws-sdk/credential-providers@3.716.0
- @aws-sdk/[email protected]
- @aws-sdk/[email protected]
- @aws-sdk/[email protected]
Expand All @@ -4182,7 +4182,7 @@ The following npm packages may be included in this product:
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected].5
- @smithy/[email protected].6
- @smithy/[email protected]
- @smithy/[email protected]
- @smithy/[email protected]
Expand Down Expand Up @@ -4416,9 +4416,9 @@ Apache License

The following npm packages may be included in this product:

- @aws-sdk/credential-provider-process@3.714.0
- @aws-sdk/credential-provider-sso@3.714.0
- @aws-sdk/credential-provider-web-identity@3.714.0
- @aws-sdk/credential-provider-process@3.716.0
- @aws-sdk/credential-provider-sso@3.716.0
- @aws-sdk/credential-provider-web-identity@3.716.0

These packages each contain the following license:

Expand Down Expand Up @@ -5318,8 +5318,8 @@ END OF TERMS AND CONDITIONS

The following npm packages may be included in this product:

- @aws-sdk/core@3.714.0
- @aws-sdk/credential-provider-http@3.714.0
- @aws-sdk/core@3.716.0
- @aws-sdk/credential-provider-http@3.716.0

These packages each contain the following license:

Expand Down Expand Up @@ -6390,20 +6390,20 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

The following npm packages may be included in this product:

- @inquirer/[email protected].3
- @inquirer/[email protected].0
- @inquirer/[email protected].1
- @inquirer/[email protected].0
- @inquirer/[email protected].3
- @inquirer/[email protected].8
- @inquirer/[email protected].0
- @inquirer/[email protected].3
- @inquirer/[email protected].3
- @inquirer/[email protected].0
- @inquirer/[email protected].3
- @inquirer/[email protected].3
- @inquirer/[email protected].3
- @inquirer/[email protected].1
- @inquirer/[email protected].4
- @inquirer/[email protected].1
- @inquirer/[email protected].2
- @inquirer/[email protected].1
- @inquirer/[email protected].4
- @inquirer/[email protected].9
- @inquirer/[email protected].1
- @inquirer/[email protected].4
- @inquirer/[email protected].4
- @inquirer/[email protected].1
- @inquirer/[email protected].4
- @inquirer/[email protected].4
- @inquirer/[email protected].4
- @inquirer/[email protected].2

These packages each contain the following license:

Expand Down Expand Up @@ -7891,9 +7891,9 @@ The following npm packages may be included in this product:
- @tokenizer/[email protected]
- [email protected]
- [email protected]
- [email protected].0
- [email protected].0
- [email protected].0
- [email protected].4
- [email protected].4
- [email protected].4
- [email protected]
- [email protected]
- [email protected]
Expand Down
4 changes: 2 additions & 2 deletions docs/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,8 @@
"dependencies": {},
"devDependencies": {
"@astrojs/check": "^0.9.4",
"@astrojs/starlight": "^0.30.2",
"astro": "^5.0.9",
"@astrojs/starlight": "^0.30.3",
"astro": "^5.1.1",
"rehype-mermaid": "^3.0.0",
"starlight-blog": "^0.16.0",
"starlight-links-validator": "^0.14.0",
Expand Down
51 changes: 51 additions & 0 deletions docs/public/schemas/llms.json
Original file line number Diff line number Diff line change
Expand Up @@ -58,13 +58,64 @@
"bearerToken": {
"type": "boolean",
"description": "Indicates if bearer token is supported"
},
"aliases": {
"type": "object",
"description": "List of model aliases for the provider",
"properties": {
"large": {
"type": "string",
"description": "Alias for large model"
},
"small": {
"type": "string",
"description": "Alias for small model"
},
"vision": {
"type": "string",
"description": "Alias for vision model"
},
"reasoning": {
"type": "string",
"description": "Alias for reasoning model"
},
"reasoning_small": {
"type": "string",
"description": "Alias for reasoning small model"
},
"long": {
"type": "string",
"description": "Alias for long model"
},
"agent": {
"type": "string",
"description": "Alias for agent model"
},
"memory": {
"type": "string",
"description": "Alias for memory model"
},
"embeddings": {
"type": "string",
"description": "Alias for embeddings model"
}
}
}
},
"additionalProperties": false,
"required": ["id", "detail"]
}
}
},
"aliases": {
"type": "object",
"additionalProperties": true,
"patternProperties": {
"^[a-zA-Z0-9:_-]+$": {
"type": "string"
}
}
},
"pricings": {
"type": "object",
"additionalProperties": false,
Expand Down
Loading
Loading