Merge branch 'main' into pangea-plugin

Author: pangea-andrest

.github/workflows/link-checker.yml

@@ -0,0 +1,50 @@
+name: Check Markdown links
+
+on:
+  push:
+    paths:
+      - '**/*.md' # Only run when markdown files change
+  pull_request:
+    branches:
+      - main
+  schedule:
+    - cron: '0 0 * * 0' # Run weekly on Sundays
+  workflow_dispatch: # Allows manual triggering
+
+jobs:
+  linkChecker:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: Link Checker
+        uses: lycheeverse/[email protected]
+        with:
+          args: --verbose --no-progress --fail './**/*.md'
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Create Issue If Failed
+        if: failure()
+        uses: actions/github-script@v6
+        with:
+          script: |
+            const title = '🔗 Broken links found in documentation';
+            const body = 'The link checker found broken links in the documentation. Please check the [workflow run](${process.env.GITHUB_SERVER_URL}/${process.env.GITHUB_REPOSITORY}/actions/runs/${process.env.GITHUB_RUN_ID}) for details.';
+
+            const existingIssues = await github.rest.issues.listForRepo({
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              labels: 'documentation,broken-links',
+            });
+
+            const issueExists = existingIssues.data.some(issue => issue.title === title);
+            if (!issueExists) {
+              await github.rest.issues.create({
+                owner: context.repo.owner,
+                repo: context.repo.repo,
+                title: title,
+                body: body,
+                labels: ['documentation', 'broken-links']
+              });
+            }

README.md

@@ -76,6 +76,7 @@ export const LAMBDA: string = 'lambda';
 export const DASHSCOPE: string = 'dashscope';
 export const X_AI: string = 'x-ai';
 export const SAGEMAKER: string = 'sagemaker';
+export const NEBIUS: string = 'nebius';
 
 export const VALID_PROVIDERS = [
   ANTHROPIC,
@@ -125,6 +126,7 @@ export const VALID_PROVIDERS = [
   DASHSCOPE,
   X_AI,
   SAGEMAKER,
+  NEBIUS,
 ];
 
 export const CONTENT_TYPES = {

cookbook/monitoring-agents/CrewAI_with_Telemetry.ipynb

@@ -786,6 +786,8 @@ export function constructConfigFromRequestHeaders(
       requestHeaders[`x-${POWERED_BY}-amzn-sagemaker-inference-component`],
     amznSagemakerSessionId:
       requestHeaders[`x-${POWERED_BY}-amzn-sagemaker-session-id`],
+    amznSagemakerModelName:
+      requestHeaders[`x-${POWERED_BY}-amzn-sagemaker-model-name`],
   };
 
   const workersAiConfig = {

src/globals.ts

@@ -54,6 +54,10 @@ export const GroqChatCompleteConfig: ProviderConfig = {
     max: 1,
     min: 1,
   },
+  tools: {
+    param: 'tools',
+    required: false,
+  },
 };
 
 export interface GroqChatCompleteResponse extends ChatCompletionResponse {}

src/handlers/handlerUtils.ts

@@ -50,6 +50,7 @@ import { DashScopeConfig } from './dashscope';
 import XAIConfig from './x-ai';
 import QdrantConfig from './qdrant';
 import SagemakerConfig from './sagemaker';
+import NebiusConfig from './nebius';
 
 const Providers: { [key: string]: ProviderConfigs } = {
   openai: OpenAIConfig,
@@ -100,6 +101,7 @@ const Providers: { [key: string]: ProviderConfigs } = {
   'x-ai': XAIConfig,
   qdrant: QdrantConfig,
   sagemaker: SagemakerConfig,
+  nebius: NebiusConfig,
 };
 
 export default Providers;

src/providers/groq/chatComplete.ts

@@ -0,0 +1,21 @@
+import { ProviderAPIConfig } from '../types';
+
+export const nebiusAPIConfig: ProviderAPIConfig = {
+  getBaseURL: () => 'https://api.studio.nebius.ai/v1',
+  headers({ providerOptions }) {
+    const { apiKey } = providerOptions;
+    return { Authorization: `Bearer ${apiKey}` };
+  },
+  getEndpoint({ fn }) {
+    switch (fn) {
+      case 'chatComplete':
+        return `/chat/completions`;
+      case 'embed':
+        return `/embeddings`;
+      case 'complete':
+        return '/completions';
+      default:
+        return '';
+    }
+  },
+};

src/providers/index.ts

@@ -0,0 +1,23 @@
+import { NEBIUS } from '../../globals';
+import {
+  chatCompleteParams,
+  embedParams,
+  completeParams,
+  responseTransformers,
+} from '../open-ai-base';
+import { ProviderConfigs } from '../types';
+import { nebiusAPIConfig } from './api';
+
+export const NebiusConfig: ProviderConfigs = {
+  chatComplete: chatCompleteParams([], {
+    model: 'Qwen/Qwen2.5-72B-Instruct-fast',
+  }),
+  embed: embedParams([], { model: 'BAAI/bge-en-icl' }),
+  complete: completeParams([], { model: 'Qwen/Qwen2.5-72B-Instruct-fast' }),
+  api: nebiusAPIConfig,
+  responseTransforms: responseTransformers(NEBIUS, {
+    chatComplete: true,
+    embed: true,
+    complete: true,
+  }),
+};

src/providers/nebius/api.ts

@@ -64,6 +64,9 @@ export const OllamaChatCompleteConfig: ProviderConfig = {
     default: 100,
     min: 0,
   },
+  tools: {
+    param: 'tools',
+  },
 };
 
 export interface OllamaChatCompleteResponse extends ChatCompletionResponse {