Complete Working Example

#!/usr/bin/env python3
"""
Private Course Assistant with RAG
Complete working example with access control and citations.
"""

import os
import chromadb
from sentence_transformers import SentenceTransformer
from dataclasses import dataclass
from typing import List, Set
from enum import Enum

# Create synthetic course materials
def create_sample_course():
    """Create sample CS101 course materials."""
    os.makedirs("courses/cs101/lectures", exist_ok=True)
    os.makedirs("courses/cs101/assignments", exist_ok=True)
    os.makedirs("courses/cs101/faqs", exist_ok=True)
    os.makedirs("courses/cs101/policies", exist_ok=True)
    
    # Lecture notes
    with open("courses/cs101/lectures/week01_intro.md", "w") as f:
        f.write("""# Week 1: Introduction to Python

## Learning Objectives
- Understand Python basics
- Set up development environment
- Write your first program

## Key Concepts
Python is a high-level, interpreted programming language.
It emphasizes code readability with significant indentation.

## Hello World
```python
print("Hello, World!")
```

## Variables
```python
name = "Alice"
age = 20
gpa = 3.8
```
""")
    
    # Assignments
    with open("courses/cs101/assignments/hw01_description.md", "w") as f:
        f.write("""# Homework 1: Python Basics

## Due Date
Friday, September 6th at 11:59 PM

## Submission
Submit via Canvas. Late submissions lose 10% per day.

## Requirements
1. Write a program that calculates the area of a circle
2. Create a function that converts temperature
3. Build a simple calculator

## Grading
- Correctness: 60%
- Code style: 20%
- Documentation: 20%
""")
    
    # FAQs
    with open("courses/cs101/faqs/submission_policy.md", "w") as f:
        f.write="""# Submission Policy FAQ

## Q: How do I submit homework?
A: Submit via Canvas. Click "Submit Assignment" on the homework page.

## Q: Can I submit late?
A: Yes, but late submissions lose 10% per day. No submissions after 3 days.

## Q: Can I work with others?
A: You may discuss concepts, but code must be your own. See academic integrity policy.

## Q: What file format should I use?
A: Submit .py files. Name them: firstname_lastname_hw01.py
""")
    
    # Policies
    with open("courses/cs101/policies/late_policy.md", "w") as f:
        f.write="""# Late Submission Policy

## Standard Policy
- Assignments due at 11:59 PM on the due date
- 10% deduction per day late
- No submissions accepted after 3 days

## Extensions
Extensions granted only for:
- Medical emergencies (with documentation)
- University-approved absences
- Contact instructor at least 24 hours before deadline

## Grade Impact
Late submissions may affect participation grade.
""")

# Run the example
if __name__ == "__main__":
    # Create sample course
    create_sample_course()
    
    # Initialize components
    embedding_model = SentenceTransformer("nomic-embed-text")
    client = chromadb.Client()
    collection = client.create_collection("course_materials")
    
    # Ingest course materials
    for root, dirs, files in os.walk("courses/cs101"):
        for file in files:
            if file.endswith(".md"):
                filepath = os.path.join(root, file)
                with open(filepath, "r") as f:
                    content = f.read()
                
                # Determine content type
                content_type = "general"
                if "lectures" in filepath:
                    content_type = "lecture"
                elif "assignments" in filepath:
                    content_type = "assignment"
                elif "faqs" in filepath:
                    content_type = "faq"
                elif "policies" in filepath:
                    content_type = "policy"
                
                embedding = embedding_model.encode(content)
                collection.add(
                    documents=[content],
                    embeddings=[embedding.tolist()],
                    ids=[filepath],
                    metadatas=[{
                        "course_id": "cs101",
                        "content_type": content_type,
                        "filename": file
                    }]
                )
                print(f"Ingested: {file}")
    
    # Query
    query = "When is homework 1 due?"
    query_embedding = embedding_model.encode([query])
    
    results = collection.query(
        query_embeddings=query_embedding.tolist(),
        n_results=3,
        where={"course_id": "cs101"}
    )
    
    print("\n=== Results ===")
    for doc, meta in zip(results["documents"][0], results["metadatas"][0]):
        print(f"\n📄 {meta['filename']} ({meta['content_type']})")
        print(f"   {doc[:200]}...")

Run It

# Save as course_assistant.py
python course_assistant.py

# Output:
# Ingested: week01_intro.md
# Ingested: hw01_description.md
# Ingested: submission_policy.md
# Ingested: late_policy.md
#
# === Results ===
# 📄 hw01_description.md (assignment)
#    # Homework 1: Python Basics
#    ## Due Date
#    Friday, September 6th at 11:59 PM...
# 📄 late_policy.md (policy)
#    # Late Submission Policy
#    ## Standard Policy
#    - Assignments due at 11:59 PM...