Chat

import StudyfetchSDK from '@studyfetch/sdk';

const client = new StudyfetchSDK({
  apiKey: 'your-api-key',
  baseURL: 'https://studyfetchapi.com',
});

const chatComponent = await client.v1.components.create({
  name: 'Biology Study Assistant',
  type: 'chat',
  config: {
    materials: ['mat-123', 'mat-456'],
    folders: ['folder-789'],
    model: 'gpt-4o-mini-2024-07-18',
    systemPrompt: 'You are a helpful biology tutor. Answer questions based on the provided materials.',
    temperature: 0.7,
    maxTokens: 2048,
    enableWebSearch: true,
    enableRAGSearch: true,
    maxSteps: 5
  }
});

console.log('Chat component created:', chatComponent._id);

const chatWithGuardrails = await client.v1.components.create({
  name: 'Math Tutor with Guardrails',
  type: 'chat',
  config: {
    materials: ['mat-789'],
    model: 'gpt-4o-mini-2024-07-18',
    enableGuardrails: true,
    guardrailRules: [
      {
        id: 'no-direct-answers',
        action: 'MODIFY',
        condition: 'Response provides direct answers to homework problems without explanation',
        description: 'Guide students to solve problems themselves',
        message: 'Modified to provide guidance instead of direct answers'
      },
      {
        id: 'academic-honesty',
        action: 'WARN',
        condition: 'User asks for test or exam answers',
        description: 'Warn about academic integrity',
        message: 'Remember: This tool is for learning, not for completing graded assignments'
      }
    ]
  }
});

const safeChat = await client.v1.components.create({
  name: 'K-12 Science Assistant',
  type: 'chat',
  config: {
    materials: ['mat-science-k12'],
    enableGuardrails: true,
    guardrailRules: [
      {
        id: 'no-medical-advice',
        action: 'BLOCK',
        condition: 'Response contains medical diagnosis or treatment advice',
        description: 'Prevent medical advice',
        message: 'I cannot provide medical advice. Please consult a healthcare professional.'
      },
      {
        id: 'age-appropriate',
        action: 'MODIFY',
        condition: 'Response contains content not suitable for K-12 students',
        description: 'Ensure age-appropriate content',
        message: 'Content adjusted for educational purposes'
      },
      {
        id: 'no-dangerous-experiments',
        action: 'BLOCK',
        condition: 'Response describes dangerous chemical reactions or experiments',
        description: 'Prevent dangerous activities',
        message: 'For safety reasons, I cannot provide instructions for this type of experiment.'
      }
    ]
  }
});

const embedResponse = await client.v1.components.generateEmbed(chatComponent._id, {
  // User tracking
  userId: 'user-456',
  groupIds: ['class-101', 'class-102'],
  sessionId: 'session-789',
  
  // Chat-specific features
  features: {
    enableWebSearch: true,
    enableHistory: true,
    enableVoice: true,
    enableFollowUps: true,
    enableComponentCreation: false,
    placeholderText: 'Ask me anything about biology...',
    enableWebSearchSources: true,
    enableImageSources: true
  },
  
  // Dimensions
  width: '100%',
  height: '600px',
  
  // Token expiry
  expiryHours: 24
});

// AI SDK format with messages array
const streamResponse = await client.v1.chat.stream({
  componentId: 'comp_123abc',
  sessionId: 'session-789',
  userId: 'user-456',
  groupIds: ['class-101', 'class-102'],
  messages: [
    { role: 'system', content: 'You are a helpful biology tutor.' },
    { role: 'user', content: 'Explain photosynthesis in simple terms' }
  ]
});

// Process the stream
for await (const chunk of streamResponse) {
  console.log(chunk.content);
}

// Custom format with single message
const customStream = await client.v1.chat.stream({
  componentId: 'comp_123abc',
  sessionId: 'session-789',
  userId: 'user-456',
  message: {
    text: 'What is cellular respiration?',
    images: [
      {
        url: 'https://example.com/cell-diagram.png',
        caption: 'Cell structure diagram',
        mimeType: 'image/png'
      }
    ]
  }
});

// React example with streaming
function ChatInterface({ componentId }) {
  const [messages, setMessages] = useState([]);
  const [streaming, setStreaming] = useState(false);

  const sendMessage = async (text) => {
    setStreaming(true);
    const userMessage = { role: 'user', content: text };
    setMessages(prev => [...prev, userMessage]);

    const assistantMessage = { role: 'assistant', content: '' };
    setMessages(prev => [...prev, assistantMessage]);

    try {
      const stream = await client.v1.chat.stream({
        componentId,
        sessionId: sessionStorage.getItem('chatSession'),
        messages: [...messages, userMessage]
      });

      for await (const chunk of stream) {
        if (chunk.type === 'content') {
          setMessages(prev => {
            const updated = [...prev];
            updated[updated.length - 1].content += chunk.content;
            return updated;
          });
        }
      }
    } finally {
      setStreaming(false);
    }
  };

  return (
    <div>
      {messages.map((msg, i) => (
        <div key={i} className={msg.role}>
          {msg.content}
        </div>
      ))}
      <input 
        onSubmit={(e) => sendMessage(e.target.value)}
        disabled={streaming}
      />
    </div>
  );
}